|
--- |
|
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 |
|
--- |