pytorch错误“\oss.cu:257: block: [0,0,0], thread: [15,0,0] Assertion `t >= 0 && t < n_classes` failed.”

错误截图:

参考这一篇详细解释:

 http://t.csdn.cn/J2XHm

通过在网上查阅资料,总结来说,是因为在使用交叉熵损失函数进行计算损失值时,发生预测值与标签值不匹配的情况所产生的错误:

不匹配情况有:

1、预测值和标签长度不一致

2、预测值或标签值不在 [0, 1] 范围内

3、如果标签中包含有字母的大小写,那么也有可能就是大小写不匹配,或者说标签格式不一致!(我自己就是这种情况)

如果发现这样的问题,进行标签数量及格式的排查就好!

猜你喜欢

转载自blog.csdn.net/qq_58611650/article/details/128162821