Debug Pytorch: ValueError: Expected more than 1 value per channel when training, got input size tor

ValueError: Expected more than 1 value per channel when training, got input size torch.Size([1, 512])

当模型中有nn.BatchNorm时，（训练时）batch不能设置为1，也不能出现样本最后一个batch可能就剩一个样本

解决方法

  • batch_size > 1
  • DataLoader中设置 drop_last=True
版权声明：本文为weixin_38812492原创文章，遵循CC 4.0 BY-SA版权协议，转载请附上原文出处链接和本声明。