Save torch model in Databricks Cluster

So the files disappear after a restart?
This seems to be a Databricks issue and I’m unfortunately not familiar with their cluster.
Could you add some persistent user folders, where you could store data or maybe mount another drive to the machine?