mbeltrami commited on
Commit
4d05f37
·
1 Parent(s): 553e5e1

Upload README.md

Browse files
Files changed (1) hide show
  1. README.md +23 -1
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
+ ---