Hi,
Thank you for your suggestion. If only I can have more bucks, I would have grab a new GPU for sure…
I’ve also tried to reduce the batch size, however the behaviors are still the same, after a few iteration the training will eventually throw OOM when I used my customized Dataloader (class MyDataset(…)), however the ImageLoader seems to be able to handle this on its own.
So, my exact interest is the point than make the ImageLoader container works without OOM while my customized method failed when working under the same conditions
So, any idea what is it?
Thanks