python code and notebooks used for AI502 Midterm Project.
- 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