Training setup for MNASNet that will get you close to "paper" numbers.
Result after 200 epochs:
- Top-1 = 73.512
- Top-5 = 91.544
This program with hardcoded settings requires 4x NVIDIA GTX 1080ti GPUs to run. Give it more epochs and it'll get even closer to paper numbers.
- PyTorch 1.0.1+
- PillowSIMD
- TensorboardX
- tqdm (for eval only)
And this PR: pytorch/vision#829 contains the MNASNet implementation so you'll need it as well.
See under checkpoints