Open KristianDukov opened 5 years ago
seems you are using a lastest version of pytorch. You can try replace the loss.data[0]
with loss.item()
dear @weigq, It conduct well on Ubuntu 18, Anaconda, Python 2.7, Pytorch 1.1 cuda. (replace the loss.data[0] with loss.item())
Running on Windows 10 Python Python 3.6.7
When trying to run the test with the given model from google drive and the dataset, I get the following error: