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

你可能感兴趣的:(python,docker,容器)