switched model saving to absolute path

apax-hub · Dec 27, 2023 · ecd2759 · ecd2759
1 parent 2f44651
commit ecd2759
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/apax/train/checkpoints.py b/apax/train/checkpoints.py
@@ -89,9 +89,9 @@ class CheckpointManager:
     def __init__(self) -> None:
         self.async_manager = checkpoints.AsyncManager()
 
-    def save_checkpoint(self, ckpt, epoch: int, path: str) -> None:
+    def save_checkpoint(self, ckpt, epoch: int, path: Path) -> None:
         checkpoints.save_checkpoint(
-            ckpt_dir=path,
+            ckpt_dir=path.resolve(),
             target=ckpt,
             step=epoch,
             overwrite=True,