Closed mingzhang96 closed 6 years ago
Hi, what version of Pytorch are you using?
If you use Pytorch > 0.3, you should modify some code. You need to use tensor.item() to convert Pytorch tensor to python data type in some lines of code instead of directly add a tensor to a variable of python data type
Hi, what version of Pytorch are you using?
using 0.4.1
@rahafaljundi I try the model with new environment as follow:
but this problem still happened. Wondering if there is still anything wrong.
I am sorry that I miss the correct environment. Now everything goes right.
I get acc=0 using the given model when running demo. Here is the output log:
Is there anything wrong with the given model _mnistnet.pth.tar?