Training of image classification models with PyTorch - View it on GitHub