fix: error in save();

WenjieDu · Nov 29, 2023 · 9acb086 · 9acb086
1 parent b0e5474
commit 9acb086
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/pypots/base.py b/pypots/base.py
@@ -272,8 +272,9 @@ def save(
                 )
             else:
                 logger.error(f"File {saving_path} exists. Saving operation aborted.")
+
         try:
-            create_dir_if_not_exist(saving_path)
+            create_dir_if_not_exist(saving_dir)
             if isinstance(self.device, list):
                 # to save a DataParallel model generically, save the model.module.state_dict()
                 torch.save(self.model.module, saving_path)