Путаница относительно размера пакета при использовании 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 изображений.