问题解决:ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1,

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就会舍去批次中多余的部分
问题解决:ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1,_第1张图片

参考链接:
https://blog.csdn.net/thy0000/article/details/123435996

你可能感兴趣的:(pytorch,零散问题,深度学习,神经网络,人工智能)