在 pytorch 中使用 DataLoader 时对批量大小的困惑答案

【问题标题】：Confusion regarding batch size while using DataLoader in pytorch在 pytorch 中使用 DataLoader 时对批量大小的困惑
【发布时间】：2020-05-26 00:11:29
【问题描述】：

我是 pytorch 的新手。我正在训练一个用于 MNIST 数据集分类的 ANN。

train_loader = DataLoader(train_data,batch_size=200,shuffle=True)

我很困惑。数据集包含 60,000 张图像，我将批量大小设置为 6000，我的模型有 30 个 epoch。是每个 epoch 只看到 6000 张图片还是每个 epoch 看到 10 批 6000 张图片？

【问题讨论】：

标签： neural-network pytorch batchsize

【解决方案1】：

对数据集迭代器的每次调用都会返回一批大小为batch_size 的图像。因此，您将有 10 个批次，直到用完所有 60000 图像。

【讨论】：