coderKyf

@HelloSeeing Hi, When I run this code ,some errors occurred. `python main.py cifar10 cifar10_LeNet ../log/cifar10_test ../data --objective one-class --lr 0.0001 --n_epochs 150 --lr_milestone 50 --batch_size 200 --weight_decay 0.5e-6 --pretrain True...

@HelloSeeing Yes, I had run this code in mnist successfully.

@HelloSeeing Thanks a lot. Torchvision is not 0.2.1. Thanks.

@HelloSeeing From class 0 to 9 are 59.51% 61.12% 62.05% 58.73% 56.57% 61.57% 57.20% 62.60% 76.44% 68.45% Maybe it would be better to change the hyperparameters.