TypeError: iteration over a 0-d tensor

作者：小丑西瓜9 | 2024-04-03 00:10:36

踩

iteration over a 0-d tensor

模型训练报错：
TypeError: iteration over a 0-d tensor
解决：
total_loss+=loss
换成如下方式
total_loss+=loss.item()
1
2
3
4
5
6

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/小丑西瓜9/article/detail/354121