我有一个在 win 10 cpu 上运行的 pytorch nn 模型。 批量大小为 42 67次迭代后，奇怪的事情发生了:batch size突然减少到28，我得到了
RuntimeError: Expected hidden[0] size (1, 28, 256), got (1, 42, 256)
最佳答案
数据集中的训练样本数量有可能不能被 42 整除吗？难道提醒的是28？
如果您的模型无法处理批量大小的在线更改，您应该考虑在 torch.utils.data.DataLoader 中设置 drop_last=True ，因此在训练期间只会处理完整的批处理。
关于machine-learning - PyTorch 批量大小在 n epoch 后突然减小，我们在Stack Overflow上找到一个类似的问题： https://stackoverflow.com/questions/56864159/