2019年12月4日
IndexError: invalid index of a 0-dim tensor. Use tensor.item() to convert a 0-dim tensor to a Python number
摘要： 报错原因分析: train_loss += loss.data[0] 是pytorch0.3.1版本代码,在0.4-0.5版本的pytorch会出现警告,不会报错,但是0.5版本以上的pytorch就会报错,总的来说是版本更新问题. 解决方法: 将datra[0]换成item（） 即可 阅读全文
posted @ 2019-12-04 21:31 闪存第一菜鸡 阅读(2006) 评论(0) 推荐(0)