Confusion regarding batch size while using DataLoader in pytorch

Question

I am new to pytorch. I am training an ANN for classification on the MNIST dataset.

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

I am confused. The dataset is of 60,000 images and I have set batch size of 6000 and my model has 30 epochs. Will every epoch see only 6000 images or will every epoch see 10 batches of 6000 images?

tourist · Accepted Answer · 2020-05-26 02:44:40Z

4

Every call to the dataset iterator will return batch of images of size batch_size. Hence you will have 10 batches until you exhaust all the 60000 images.

edited May 26, 2020 at 2:44

answered May 26, 2020 at 2:38

tourist

4,3537 gold badges28 silver badges48 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

Confusion regarding batch size while using DataLoader in pytorch

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related