pytorch_impementation FCN (basic pipeline) CNN (basic pipeline) LeNet VGG googlenet build custom datasets data augmentation using torchvision save load models transfer learning and fine tuning using learning rate scheduler weight initialization @ thanks to aladdin