IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number

[roshan-gopalakrishnan]

Traceback (most recent call last): File "main.py", line 166, in train(epoch) File "main.py", line 127, in train avg_loss += loss.data[0] IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number

[Eric-mingjie]

Use torch v0.3.1.

[songheony]

If you want to use Pytorch >= 0.4.1, you have to edit some code. Here is the source code I converted https://github.com/songheony/rethinking-network-pruning. I've tested it with Pytorch 1.2, CUDA 10

[tejalal]

avg_loss += loss.data[0]

Replace avg_loss += loss.data[0] with avg_loss += loss.item()