Little bug in main.py - Githubissues

ZhouYanzhao / ORN

Oriented Response Networks, in CVPR 2017

http://yzhou.work/ORN

BSD 3-Clause "New" or "Revised" License

223 stars 51 forks source link

Little bug in main.py #12

Open ymcasky opened 6 years ago

ymcasky commented 6 years ago

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

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

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 : )