Assertion `cur_target >= 0 && cur_target < n_classes’ failed.

爱心觉罗

讲一个我们在分类训练中经常遇到的问题：

一般来说在我们网络中输出的种类数和你label设置的种类数量不同的时候就会出现这个错误。

但是，Pytorch有个要求，在使用CrossEntropyLoss这个函数进行验证时label必须是以0开始的：

假如我这样:

1. self.classes = [0, 1, 2, 3]

复制代码

我的种类有四类，分别是0.1.2.3，这样就没有什么问题，但是如果我写成：

1. self.classes = [1, 2, 3, 4]

复制代码

这样就会报错

shaoheshaohe

嗯
类别代码必须哦那个0开始

shaoheshaohe

这个事
初学者
需要知道
才不会犯错误

shaoheshaohe

一般来说，许多数组型的
结构
都是从0开始计数
python如此