一般来说在网络中输出的种类数和你label设置的种类数量不同的时候就会出现这个错误。
但是,Pytorch有个要求,在使用CrossEntropyLoss这个函数进行验证时label必须是以0开始的
所以在使用pytoch时出现上述错误,又使用
criterion = torch.CrossEntropyLoss()
那么就要观察文本中训练集和测试集的标签是否从0开始。
根据该博客,我debug看了标签,是从0开始的 ,目前还没有解决
一般来说在网络中输出的种类数和你label设置的种类数量不同的时候就会出现这个错误。
但是,Pytorch有个要求,在使用CrossEntropyLoss这个函数进行验证时label必须是以0开始的
所以在使用pytoch时出现上述错误,又使用
criterion = torch.CrossEntropyLoss()
那么就要观察文本中训练集和测试集的标签是否从0开始。
根据该博客,我debug看了标签,是从0开始的 ,目前还没有解决