当运行下面代码时出现下面的错误
if iteration%10 == 0:
print("===> Epoch[{}]({}/{}): Loss: {:.10f}".format(epoch, iteration, len(training_data_loader), loss.data[0]))
修改建议:
去掉loss.data[0]后面的[0]。
if iteration%10 == 0:
print("===> Epoch[{}]({}/{}): Loss: {:.10f}".format(epoch, iteration, len(training_data_loader), loss.data))
原因:loss是0维张量
创建一个Tensor默认是不可导的,即 .requeires_grad 属性等于false。
在pytorch0.4版本之后,Variable和Tensor进行的合并。loss.data直接输出tensor值,不输出tensor的梯度信息。比如:
文章来源:https://www.toymoban.com/news/detail-430982.html
loss是0维张量,因此在输出loss.data的时候维度为0,可以用loss.data.size()测试。可以利用loss.item()将张量转换为python number。文章来源地址https://www.toymoban.com/news/detail-430982.html
到了这里,关于IndexError: invalid index of a 0-dim tensor. Use `tensor.item()` in Python的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!