Sample command does not work

[Chao1155]

The sample command in readme does not work directly.

python main.py --model BNN_vgg_cifar10 --save BNN_cifar10 --dataset cifar10

There is some problems in flag settings.

[Krylol94]

I have the same issue. Did you find the solution?

[Viscoo]

I have the issue as follows: AssertionError: no model file named: models\BNN_vgg_cifar10.py

[Krylol94]

This problems arises because of the wrong name, write "python main.py --model BNN_cifar10 --save BNN_cifar10 --dataset cifar10" instead