zl程序教程

您现在的位置是:首页 >  其他

当前栏目

成功解决Assertion `cur_target >= 0 && cur_target < n_classes‘ failed.

ampFailed 解决 成功 Target classes Assertion
2023-09-11 14:20:47 时间

成功解决Assertion `cur_target >= 0 && cur_target < n_classes’ failed.

解决方案

1,一般来说在网络中输出的种类数和你label设置的种类数量不同的时候就会出现这个错误。
举例来说,你的标签是10类,但是最后全连接层的输出节点个数不是10,就会报错,改一下就好

2,Pytorch有个要求,在使用CrossEntropyLoss这个函数进行验证时label必须是以0开始
所以在使用pytoch时出现上述错误,
在使用criterion = torch.CrossEntropyLoss() 时
那么就要观察文本中训练集和测试集的标签是否从0开始。

3,注意在windows系统下,num_workers要设置为0,即:num_workers=0

点个免费的赞呗,求支持!