Solved by Oli in post #2 
              
              
                Hi, 
I can’t say that I’m an expert in this. Seems like an annoying issue. Some things that you can try. 

Make sure the networks are transferred to the CPU before saving
Is your loss dictionaries of type tensor or numpy? If they are tensor, extract with loss_tensor.numpy()
Saving the dictionaries o…