首页 > 其他 > 详细

unable to get repr for class ‘torch.tensor’

时间:2021-06-19 09:18:22      阅读:21      评论:0      收藏:0      [点我收藏+]

1.调试程序发现计算loss的时候出现如题所示错误,导致后续程序报CUDA的错

nn.CrossEntropyLoss(),我设置ignore_index为0,mask设为0,1,2,做2分类,就会这样

 

2.将ignore_index改为-1,计算loss的时候把mask=mask-1,就能正常分类

 

不知道为啥会出现这种情况

unable to get repr for class ‘torch.tensor’

原文:https://www.cnblogs.com/zmbreathing/p/loss_ignoreindex.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!