Model | Total Parameters | Foward/Backward Pass Size (MB) | Params Size (MB) | Estimated Total Size (MB) | Top-1 Train Accuracy (%) | Top-1 Test Accuracy (%) |
---|---|---|---|---|---|---|
Vision Transformer (ViT) | 86,415,592 | 408.54 | 329.65 | 738.77 | 99.4 | 92.5 |
ResNet50 | 25,557,032 | 286.57 | 97.49 | 384.64 | 95.8 | 86.6 |
EfficientNet | 5,288,548 | 173.44 | 20.17 | 194.19 | 98.3 | 90.2 |