Upload README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,25 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
license: mit
|
| 3 |
-
---
|
|
|
|
| 1 |
+
# micromind checkpoints for MNIST
|
| 2 |
+
|
| 3 |
+
This repository contains checkpoints for the MNIST dataset for the following networks:
|
| 4 |
+
|
| 5 |
+
| Model | Top 1 Accuracy | Top 5 Accuracy |
|
| 6 |
+
| ------------------ |---------------- | -------------- |
|
| 7 |
+
| `PhiNet(alpha=0.5, beta=1, t_zero=6, num_layers=4, resolution=28)` | 98.97% | 99.99% |
|
| 8 |
+
| `PhiNet(alpha=0.75, beta=1, t_zero=6, num_layers=5, resolution=28)` | 98.94% | 99.98% |
|
| 9 |
+
| `PhiNet(alpha=0.35, beta=1, t_zero=6, num_layers=7, resolution=28)` | 98.83% | 100% |
|
| 10 |
+
|
| 11 |
+
To download and use this repo:
|
| 12 |
+
|
| 13 |
+
```
|
| 14 |
+
from micromind import PhiNet
|
| 15 |
+
model = PhiNet.from_pretrained("MNIST", alpha=0.5, beta=1.0, t_zero=6, num_layers=4, num_classes=10, resolution=28)
|
| 16 |
+
```
|
| 17 |
+
|
| 18 |
+
## Authors
|
| 19 |
+
|
| 20 |
+
- [@fpaissan](https://www.github.com/fpaissan)
|
| 21 |
+
- [@matteobeltrami](https://www.github.com/matteobeltrami)
|
| 22 |
+
|
| 23 |
---
|
| 24 |
license: mit
|
| 25 |
+
---
|