nn.BatchNorm2d() 这个函数在训练时只能输入batch size >1 的数据。 train 阶段 因此如果在最后一个Batch只有一个数据时应把该batch丢弃。 在torch.utils.data.Dataloader中设置参数: drop_last=True 不足batch大小的最后部分舍去 eval阶段 在评价模型时如果需要输入单张图片,可以在前面加上: net.eval