Hi,
I think we might need some more information before we can understand where the problem is.

are you using multiple GPUs to train the model?
can you remove the metric_logger logging, and just iterate over the dataloader

Those are the two things that came to my mind. I haven’t experienced this OOM before, so I’m not sure what else it could be for now