Zalando's MNIST replacement.
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
.gitignore Finished porting to 0.4 Jun 25, 2018
LICENSE Initial commit Aug 28, 2017
README.md Some changes Jun 25, 2018
model.py added mnist Aug 28, 2017
train.py Some changes Jun 25, 2018
utils.py Setting up Aug 28, 2017

README.md

Zalando's MNIST fashion replacement

Zalando recently released an MNIST replacement. The issues with using MNIST are known but you can read about the dataset and their motivation here.

Training

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

Results

FashionSimpleNet ResNest18
MNIST 0.994 0.994
FashionMNIST 0.923 0.920