From 9b9a7f2d0043b54aac0119478bc183e3539b1a7c Mon Sep 17 00:00:00 2001 From: dinhanhx Date: Sun, 29 Oct 2023 10:04:34 +0000 Subject: [PATCH] Update hardcode path --- kit/extract_lightning_ckpt.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/kit/extract_lightning_ckpt.py b/kit/extract_lightning_ckpt.py index 5c7381c..736d375 100644 --- 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")