Confused about Batch_size

pytorch / ignite

High-level library to help with training and evaluating neural networks in PyTorch flexibly and transparently.

https://pytorch-ignite.ai

BSD 3-Clause "New" or "Revised" License

4.5k stars 608 forks source link

Confused about Batch_size #3233

Closed AnnaTrainingG closed 3 months ago

AnnaTrainingG commented 3 months ago

🐛 Bug description

HI, when I use the auto_dataloder, I really confused about the code in the image, If I update my batch_size from 1 to 16, and world_size still is 8, Need I set batch_size = batch_size world_size. eg: auto_dataloder( batch_size = batch_size world_size ... ? thanks

vfdev-5 commented 3 months ago

Closing this one as a duplicate of https://github.com/pytorch/ignite/issues/3234. Let's keep discussions in another ticket