RuntimeError: Assertion cur_target = 0 cur_target 小于 n_classes failed. at /opt/conda/conda-bld
Assertion `cur_target >= 0 && cur_target < n_classes’ failed
一般来说在网络中输出的种类数和你label设置的种类数量不同的时候就会出现这个错误。
但是，Pytorch有个要求，在使用CrossEntropyLoss这个函数进行验证时label必须是以0开始的
所以在使用pytoch时出现上述错误，又使用
criterion = torch.CrossEntropyLoss()
那么就要观察文本中训练集和测试集的标签是否从0开始。
本文来自互联网用户投稿，文章观点仅代表作者本人，不代表本站立场，不承担相关法律责任。如若转载，请注明出处。 如若内容造成侵权/违法违规/事实不符，请点击【内容举报】进行投诉反馈！