搜到的解决方法大致都是 cuda/torch/python版本匹配batch_size调整 尝试后并不能解决问题。。 在stackoverflow上看到这篇回答: 因为我重写了复现模型的数据读取方法,然后尝试 匹配label和model的size修改多分类label格式(从0开始,或是从1开始) 然后问题解决了,供参考