sergimassabates1:

I can only save it on the Cluster CPU (which is unnaccessible)


I’m not sure I understand this sentence correctly.
Are you able to access some folders on this machine?
If so, you should be able to store the state_dict there (similar to your script files).