Minor fixes in train.py (acc_wvi and torch.cuda.is_available())

naver / sqlova

Apache License 2.0

632 stars 168 forks source link

Minor fixes in train.py (acc_wvi and torch.cuda.is_available()) #19

Closed hanrelan closed 5 years ago

hanrelan commented 5 years ago

This pull request fixes 2 issues: 1) there's a bug in the calculation of acc_wvi for the training loop 2) some of the torch.cuda.is_available() calls are broken

whwang299 commented 5 years ago

Thanks!