Путаница относительно размера пакета при использовании DataLoader в pytorch

Question

Я новичок в pytorch. Я обучаю ИНС для классификации в наборе данных MNIST.

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

Я запутался. Набор данных состоит из 60 000 изображений, и я установил размер пакета 6000, и моя модель имеет 30 эпох. Будет ли каждая эпоха видеть только 6000 изображений или каждая эпоха будет видеть 10 групп по 6000 изображений?

Answer 1

Каждый вызов итератора набора данных возвращает пакет изображений размером batch_size. Следовательно, у вас будет 10 пакетов, пока вы не исчерпаете все 60000 изображения.