爱心觉罗 发表于 2018-10-15 17:35:01

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

讲一个我们在分类训练中经常遇到的问题：一般来说在我们网络中输出的种类数和你label设置的种类数量不同的时候就会出现这个错误。但是，Pytorch有个要求，在使用CrossEntropyLoss这个函数进行验证时label必须是以0开始的：假如我这样:self.classes = 我的种类有四类，分别是0.1.2.3，这样就没有什么问题，但是如果我写成：self.classes = 这样就会报错

shaoheshaohe 发表于 2019-2-9 19:00:13

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

shaoheshaohe 发表于 2019-2-9 19:00:28

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

shaoheshaohe 发表于 2019-2-9 19:01:00

一般来说，许多数组型的
结构
都是从0开始计数
python如此
页: [1]
查看完整版本: Assertion `cur_target >= 0 && cur_target < n_classes’ failed.