search

andrewliao11 / dni.pytorch

Implement Decoupled Neural Interfaces using Synthetic Gradients in Pytorch
118 stars 42 forks source link

Command from README not working #6

Open bgavran opened 6 years ago

bgavran commented 6 years ago

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 commented 6 years ago

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