Merge pull request #7 from microsoft/mallamanis/simplify-saving

Simplify model saving, when parent folder doesn't exist.

Author: Miltos

ptgnn/baseneuralmodel/abstractneuralmodel.py

@@ -1,6 +1,7 @@
 from typing_extensions import final
 
 import gzip
+import os
 import torch
 from abc import ABC, abstractmethod
 from concurrent import futures
@@ -151,6 +152,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)