'CUDA error: out of memory' after several epochs

Maybe don’t use

pin_memory()