MNIST / README.md
mbeltrami's picture
Update README.md
d983795
metadata
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


license: mit