Error when executing the demo command

[kanezaki]

Error when executing the demo command

[dallaqua]

Hi,
I tried to execute the demo command on README and get the below errors:

python demo.py --input images/101027.jpg
Traceback (most recent call last):
  File "demo.py", line 118, in <module>
    print (batch_idx, '/', args.maxIter, ':', nLabels, loss.data[0])
IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number

I'm not familiar with tensor though. Could you please suggest how to modify the code to solve this problem?
My system is Windows 10. Python version 3.7.3. PyTorch==1.2+cpu, opencv-python==4.1.1.26, scikit-image==0.15.0.

I had this same issue and I found that it happened because of PyTorch version. Mine is 1.2.0.
In demo.py you can find:

print (batch_idx, '/', args.maxIter, ':', nLabels, loss.data[0])

# for pytorch 1.0
#print (batch_idx, '/', args.maxIter, ':', nLabels, loss.item())

So what worked for me was to comment line 118 and uncomment line 121 - #print (batch_idx, '/', args.maxIter, ':', nLabels, loss.item())

[kanezaki]

I uncommented that line now. Thanks @dallaqua!