pytorch报错:UserWarning: invalid index of a 0-dim tensor. This will be an error in PyTorch 0.5.。。原因

UserWarning: invalid index of a 0-dim tensor. This will be an error in PyTorch 0.5. Use tensor.item() to convert a 0-dim tensor to a Python number

当报这个错的一般会有两个原因,一个是这种写法:

loss.data[0]   

应该改成loss.item()

还有就是这种写法:

im = Variable(im.cuda(),volatile=True)

应该改成im = Variable(im.cuda()) 

 

发布了246 篇原创文章 · 获赞 73 · 访问量 23万+
展开阅读全文

没有更多推荐了,返回首页

©️2019 CSDN 皮肤主题: 大白 设计师: CSDN官方博客

分享到微信朋友圈

×

扫一扫,手机浏览