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.
Additional figure 1: Comparison of Resnet50 & MobileNetV2
Dataset = CIFAR100
Batch size = 4096
lr = 5e-2
num_epochs = 1000
Optimizer = Adam
HW = RTX 2080Ti \
- 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