1.调试程序发现计算loss的时候出现如题所示错误，导致后续程序报CUDA的错
nn.CrossEntropyLoss()，我设置ignore_index为0，mask设为0,1,2，做2分类，就会这样
2.将ignore_index改为-1，计算loss的时候把mask=mask-1，就能正常分类
不知道为啥会出现这种情况
免责声明！
本站转载的文章为个人学习借鉴使用，本站对版权不负任何法律责任。如果侵犯了您的隐私权益，请联系本站邮箱yoyou2525@163.com删除。
猜您在找
torch.Tensor和torch.tensor的区别
torch.tensor(),torch.Tensor()
torch.Tensor.view (Python method, in torch.Tensor)
torch.tensor拼接与list(tensors)
torch.Tensor和numpy.ndarray
torch.Tensor中的clone()方法
torch.Tensor文档学习笔记
torch.tensor拼接与list(tensors)
torch.Tensor 与 numpy.ndarray的相互转化
torch.tensor默认requires_grad=False