Implementation and light training examples1 for popular Deep Learning models in Pytorch from scratch.
- AlexNet: trained on CIFAR-10, paper
- VGG-16: trained on Covid-19 chest x-ray dataset from kaggle, paper
- InceptionV1 (GoogLeNet): trained on brain tumor MRI dataset from kaggle, paper
- ResNet-50: trained on Animal-10 dataset from kaggle, paper
- EfficientNet: trained on car damage severity from kaggle, paper
- Vision Transformer: trained on human action recognition from kaggle, paper
Footnotes
-
PS: The current goal of this project is to present simple examples written from scratch. It doesn't dive deep into optimizing the performance or analyzing the training (At least for now). โฉ