Merge branch 'dev' into epoch_jit

apax-hub · Jan 11, 2024 · 008c3a8 · 008c3a8
2 parents e828e52 + 55b96b6
commit 008c3a8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/apax/md/nvt.py b/apax/md/nvt.py
@@ -271,7 +271,7 @@ def body_fn(i, state):
                     )
                 ckpt = {"state": state, "step": step}
                 checkpoints.save_checkpoint(
-                    ckpt_dir=ckpt_dir,
+                    ckpt_dir=ckpt_dir.resolve(),
                     target=ckpt,
                     step=step,
                     overwrite=True,