python code, notebooks and Images used for AI502 Midterm Project.
Figure 2: Comparison of test loss and train accuracy for Batch size increasing and Learning rate decaying.
Figure 4: Comparison of Adam, RAdam and SGD for MNIST classifier
- MobileNetV2: Inverted residuals and linear bottlenecks
ResNet : https://pytorch.org/docs/stable/torchvision/models.html
MobileNetV2 : https://github.com/pytorch/vision/tree/master/torchvision/models