存在的问题
train.py:None (train.py)
train.py:116: in
print(“Epoch [%d/%d], Iter [%d/%d] Loss: %.4f” %(epoch+1, total_epoch, i+1, len(train_loader), loss.data[0]))
E IndexError: invalid index of a 0-dim tensor. Use tensor.item() in Python or tensor.item<T>() in C++ to convert a 0-dim tensor to a number
collected 0 items / 1 error
解决方案
print(“Epoch [%d/%d], Iter [%d/%d] Loss: %.4f” %(epoch+1, total_epoch, i+1, len(train_loader), loss.data[0]))
修改为
print(“Epoch [%d/%d], Iter [%d/%d] Loss: %.4f” %(epoch+1, total_epoch, i+1, len(train_loader), loss.item()))
版权声明：本文为yayayamaomaoya原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。