1.调试程序发现计算loss的时候出现如题所示错误,导致后续程序报CUDA的错
nn.CrossEntropyLoss(),我设置ignore_index为0,mask设为0,1,2,做2分类,就会这样
2.将ignore_index改为-1,计算loss的时候把mask=mask-1,就能正常分类
不知道为啥会出现这种情况
2024-01-23 08:44:28
1.调试程序发现计算loss的时候出现如题所示错误,导致后续程序报CUDA的错
nn.CrossEntropyLoss(),我设置ignore_index为0,mask设为0,1,2,做2分类,就会这样
2.将ignore_index改为-1,计算loss的时候把mask=mask-1,就能正常分类
不知道为啥会出现这种情况
下一篇:消息认证码