MNIST / README.md
mbeltrami's picture
Update README.md
d983795
---
license: mit
datasets:
- mnist
language:
- en
pipeline_tag: image-classification
---
# micromind checkpoints for MNIST
This repository contains checkpoints for the MNIST dataset for the following networks:
| Model | Top 1 Accuracy | Top 5 Accuracy |
| ------------------ |---------------- | -------------- |
| `PhiNet(alpha=0.5, beta=1, t_zero=6, num_layers=4, resolution=28)` | 98.96% | 100% |
| `PhiNet(alpha=0.75, beta=1, t_zero=6, num_layers=5, resolution=28)` | 99.03% | 99.98% |
| `PhiNet(alpha=0.35, beta=1, t_zero=6, num_layers=7, resolution=28)` | 98.72% | 99.99% |
| `PhiNet(alpha=0.25, beta=1, t_zero=6, num_layers=7, resolution=28)` | 98.84% | 99.99% |
| `PhiNet(alpha=0.25, beta=1, t_zero=5, num_layers=7, resolution=28)` | 98.76% | 99.97% |
To download and use this repo:
```
from micromind import PhiNet
model = PhiNet.from_pretrained("MNIST", alpha=0.5, beta=1.0, t_zero=6, num_layers=4, num_classes=10, resolution=28)
```
## Authors
- [@fpaissan](https://www.github.com/fpaissan)
- [@matteobeltrami](https://www.github.com/matteobeltrami)
---
license: mit
---