Why CUDA runs out of memory when calling torch.save?

4

+1. I had the same problem and moving the model to cpu before saving also worked.