Solved by tengerye in post #3 
              
              
                Hi, 
Thank you for the kind reply. 

I used one gpu. But even if I used CPU, the situation happens;
I did try to remove metric_logger, but it still happens. I doubt if the model itself caused it.

The original tutorial has such few epochs to trigger the OOM, but I believe you can easily repeat my si…