update train.py

allenai · Jul 31, 2023 · 517d4b0 · 517d4b0
1 parent 9f75dd6
commit 517d4b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/scripts/train.py b/scripts/train.py
@@ -191,7 +191,7 @@ def main(cfg: TrainConfig) -> None:
 
         if cfg.load_path is not None:
             log.info(f"Loading checkpoint from {cfg.load_path}...")
-            trainer.restore_checkpoint(Path(cfg.load_path))
+            trainer.restore_checkpoint(cfg.load_path)
             log.info("Checkpoint successfully loaded")
 
         if cfg.force_save_unsharded: