络腮胡菲菲 发表于 2024-7-18 22:10:30

【pytorch】thread: [16,0,0] Assertion `t >= 0 && t < n_classes` fail

报错信息:C:\cb\pytorch_1000000000000\work\aten\src\ATen\native\cuda\Loss.cu:250: block: , thread: Assertion `t >= 0 && t < n_classes` failed
可能缘故原由:标签label或者猜测的结果pred,超出了数据的范围,好比标签内里只有0-15的数字,但是pred中出现19这个数字,那么在交织熵损失计算中就会报错。

免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。
页: [1]
查看完整版本: 【pytorch】thread: [16,0,0] Assertion `t >= 0 && t < n_classes` fail