Little bug in main.py

[ymcasky]

Hi, ZhouYanzhao

Thanks for your sharing code!

line 163 should change to :
if test_acc > best_test_acc[0]: => if test_acc.float() > best_test_acc[0]:

[ZhouYanzhao]

Hi, @ymcasky
Thanks! The code does look a bit dated. I'm going to give it an upgrade next month to support the latest version of pytorch.

[ymcasky]

Thank you !!
Looking for your update with full function XD !
I think change from
test_acc = 100. * correct / len(test_loader.dataset) to
test_acc = 100. * correct.float() / len(test_loader.dataset) is better : )