Skip to content

Commit 6d6476b

Browse files
committed
One idea for resume fix
1 parent 2a20607 commit 6d6476b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

olmocr/train/train.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -188,7 +188,7 @@ def main():
188188
checkpoint_dirs.sort(key=lambda x: int(x.split("-")[1]))
189189
latest_checkpoint = os.path.join(full_output_dir, checkpoint_dirs[-1])
190190
logger.info(f"Found existing checkpoint: {latest_checkpoint}")
191-
found_resumable_checkpoint = True
191+
found_resumable_checkpoint = latest_checkpoint
192192
else:
193193
logger.info("No existing checkpoints found in output directory")
194194

0 commit comments

Comments
 (0)