【pytorch】thread: [16,0,0] Assertion `t ＞= 0 && t ＜ n_classes` failed

作者：代码探险家 | 2024-07-20 23:29:52

踩

报错信息：C:\cb\pytorch_1000000000000\work\aten\src\ATen\native\cuda\Loss.cu:250: block: [0,0,0], thread: [16,0,0] Assertion `t >= 0 && t < n_classes` failed

可能原因：标签label或者预测的结果pred，超出了数据的范围，比如标签里面只有0-15的数字，但是pred中出现19这个数字，那么在交叉熵损失计算中就会报错。

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