This repo is archived.
Zalando recently released an MNIST replacement. The issues with using MNIST are known but you can read about the dataset and their motivation here.
python train.py
--model # specify model, (FashionSimpleNet, resnet18)
--patience # early stopping
--batch_size # batch size
--nepochs # max epochs
--nworkers # number of workers
--seed # random seed
--data # MNIST or FashionMNIST
FashionSimpleNet | ResNest18 | |
---|---|---|
MNIST | 0.994 | 0.994 |
FashionMNIST | 0.923 | 0.920 |