mbeltrami commited on
Commit
6004bbc
·
1 Parent(s): e2befdc

Upload README.md

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