machine-learning - PyTorch 批量大小在 n epoch 后突然减小

Question

我有一个在 win 10 cpu 上运行的 pytorch nn 模型。批量大小为 4267次迭代后，奇怪的事情发生了:batch size突然减少到28，我得到了

RuntimeError: Expected hidden[0] size (1, 28, 256), got (1, 42, 256)

Answer 1

数据集中的训练样本数量有可能不能被 42 整除吗？难道提醒的是28？

如果您的模型无法处理批量大小的在线更改，您应该考虑在 torch.utils.data.DataLoader 中设置 drop_last=True ，因此在训练期间只会处理完整的批处理。