This gets correctly saved and can be loaded but when cluster is restarted it disappears:

torch.save(model.state_dict(), ‘filename.pt’)

Andthis returns de error mentioned before:

torch.save(model.state_dict(), ‘dbfs:/existingfilepath/filename.pt’)