Links for train-pytorch