问题分析 设置的batch_size=2,但是样本数除以batch_size是无法整除的。由于整个网络使用的BatchNorm操作都是需要多于一个数据计算平均值,可能是这里造成了错误。 解决方法 找到网络的 DataLoader 部分,将 drop_last 设置为 True。