MNASNet trainer

Training setup for MNASNet that will get you close to "paper" numbers.

Result after 200 epochs:

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.

Requirements:

And this PR: pytorch/vision#829 contains the MNASNet implementation so you'll need it as well.

Name		Name	Last commit message	Last commit date
Latest commit History 52 Commits
checkpoints		checkpoints
training_logs		training_logs
.style.yapf		.style.yapf
LICENSE		LICENSE
README.md		README.md
eval.py		eval.py
human_readable.py		human_readable.py
imagenet.py		imagenet.py
log.py		log.py
metrics.py		metrics.py
performance.py		performance.py
tensorboard.py		tensorboard.py
train.py		train.py
trainer.py		trainer.py