摘要： 1.调试程序发现计算loss的时候出现如题所示错误，导致后续程序报CUDA的错 nn.CrossEntropyLoss()，我设置ignore_index为0，mask设为0,1,2，做2分类，就会这样 2.将ignore_index改为-1，计算loss的时候把mask=mask-1，就能正常分类 阅读全文