Simplify saving, when parent folder doesn't exist.

microsoft · Nov 27, 2020 · ab3e309 · ab3e309
1 parent 0ba8308
commit ab3e309
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ptgnn/baseneuralmodel/abstractneuralmodel.py b/ptgnn/baseneuralmodel/abstractneuralmodel.py
@@ -151,6 +151,7 @@ def build_neural_module(self) -> TNeuralModule:
 
     # region Saving/Loading
     def save(self, path: Path, model: TNeuralModule) -> None:
+        os.makedirs(os.path.dirname(str(path)), exist_ok=True)
         with gzip.open(path, "wb") as f:
             torch.save((self, model), f)