ValueError: Expected input batch_size (32) to match target batch_size (16).

错误展示:

ValueError: Expected input batch_size (32) to match target batch_size (16).

刚开始看到这个BUG,逻辑上有点想不通,但后来想到有可能是数据集本身问题时,我就顺带手看了下数据集的size()
果不其然,看图:
ValueError: Expected input batch_size (32) to match target batch_size (16)._第1张图片
最终判断,应该是数据集的部分数据的label值丢失造成的。
两种解决方案:

  1. 重新选择数据集,或者人工拍错;
  2. 适当修改代码解决BUG;
    坦诚讲:由于我比较懒,所以选择了后者;
    调整后的代码如下:
    ValueError: Expected input batch_size (32) to match target batch_size (16)._第2张图片
    1个epoch测试一下,问题解决:
    ValueError: Expected input batch_size (32) to match target batch_size (16)._第3张图片

你可能感兴趣的:(ValueError,to,match)