neural-network - 在 pytorch 中使用 DataLoader 时对批量大小的困惑

Question

我是pytorch的新手。我正在训练 ANN 对 MNIST 数据集进行分类。

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

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

Answer 1

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