RaLo4:

First up I would recommend using square images if possible. For example 224 x 224.
On how to train on your gpu with a specific batch size:
When defining a dataloader you can specify a batch size like so:


Thank you so much, I did so. But still can’t finish one epochs. I’m wondering how ImageNet runs on gpu. Because 124G data can be ran on one gpu.