模型训练报错:
TypeError: iteration over a 0-d tensor
解决:
total_loss+=loss
换成如下方式
total_loss+=loss.item()
TypeError: iteration over a 0-d tensor
最新推荐文章于 2024-07-05 11:51:13 发布
模型训练报错:
TypeError: iteration over a 0-d tensor
解决:
total_loss+=loss
换成如下方式
total_loss+=loss.item()