Update hardcode path

dinhanhx · Oct 29, 2023 · 9b9a7f2 · 9b9a7f2
1 parent b2a6fd5
commit 9b9a7f2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/kit/extract_lightning_ckpt.py b/kit/extract_lightning_ckpt.py
@@ -5,8 +5,8 @@
 from exp.run_pretrain import Wrapper
 
 if __name__ == "__main__":
-    ckpt_folder = Path("logs/lightning_logs/version_0/checkpoints")
-    ckpt_file = ckpt_folder.joinpath("epoch=1-step=18000.ckpt")
+    ckpt_folder = Path("pls-logs/lightning_logs/version_4/checkpoints/")
+    ckpt_file = ckpt_folder.joinpath("epoch=1-step=147000.ckpt")
     assert ckpt_file.is_file()
     wrapper = Wrapper.load_from_checkpoint(checkpoint_path=ckpt_file)
     pt_file = ckpt_folder.joinpath("imagetext-base.pt")