Command from README not working

[bgavran]

Hello, after cloning the repo, trying to run the command

python main.py --model_type mlp

and fixing the print command in train.py (it was the python2 version of print) I get the following error:

Traceback (most recent call last): File "main.py", line 20, in assert args.dataset != 'cifar10' or args.model_type != 'mlp' AssertionError

I'm running Python 3.6 and the 0.2.0 version of Pytorch, but this doesn't seem to be related to any of those.

[lkhoad]

main.py say that # do not support using mlp to trian cifar Assuming you want to train mnist with mlp, you should input python3 main.py --model_type mlp --dataset mnist