From 6d6476b31a44fa11f28f24a4890ff99fb8c8e997 Mon Sep 17 00:00:00 2001 From: Jake Poznanski Date: Wed, 2 Jul 2025 01:33:37 +0000 Subject: [PATCH] One idea for resume fix --- olmocr/train/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/olmocr/train/train.py b/olmocr/train/train.py index 324b441..212cf93 100644 --- a/olmocr/train/train.py +++ b/olmocr/train/train.py @@ -188,7 +188,7 @@ def main(): checkpoint_dirs.sort(key=lambda x: int(x.split("-")[1])) latest_checkpoint = os.path.join(full_output_dir, checkpoint_dirs[-1]) logger.info(f"Found existing checkpoint: {latest_checkpoint}") - found_resumable_checkpoint = True + found_resumable_checkpoint = latest_checkpoint else: logger.info("No existing checkpoints found in output directory")