ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 1280])
出现这个错误的根本原因是输入的数据只有一个数据点,但是BatchNorm操作必须超过一个数据计算平均值,可能的原因和解决方式如下:
把batch_size的值改为大于1的数
如果batch_size的值是大于1的还出现这个错误,可能是数据集的总数除以batch_size的值不能整除,但余下的数据只有一个。我们可以在dataloader中把drop_last 参数设置为True,这样dataloader就会舍去批次中多余的部分
参考链接:
https://blog.csdn.net/thy0000/article/details/123435996