pytorch-playground icon indicating copy to clipboard operation
pytorch-playground copied to clipboard

Published 20 hours ago • verified publisher icon aaron-xichen

The accuracy error

Open louvinci opened this issue 4 years ago • 1 comments

Thank you for you share, but when i run like below, i get result :acc1=0.00 acc5=0.00 ''python quantize.py --type cifar10 --quant_method linear --param_bits 16 --fwd_bits 16 --bn_bits 16 --gpu 0 --ngpu 1 --input_size 32''

The first time I run, I get the following error, ''...not supported on CUDAType for Long''

then I change the data type to float, and then I get the above result. Thank you or your help.

louvinci avatar Aug 24 '20 12:08 louvinci

I sloved this problem. .float() should also be added to the eval_model() function

louvinci avatar Aug 24 '20 14:08 louvinci