成功解决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
点个免费的赞呗,求支持!