weixin_52599014的博客

原创 Assertion `t ＞= 0 && t ＜ n_classes` failed.报错解决办法

之前在做自定义的dataset的时候其实也遇到过，但是没太在意，现在又遇到了所以记录一下，网上讲解有挺多，说一下我遇到的这种。除了上面这个断言还会有如下的报错提示。可以看出出错的地方基本上就是在损失函数那个地方。，因为交叉熵损失的y_hat（真实标签）的取值范围是[0,类别数-1],从0开始算的到类别数-1就是你的类别数。如果不是的话就会报上面标题的错误Assertionfailed.注意这里t是在**[0,num_class)**之间的。