We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2a20607 commit 6d6476bCopy full SHA for 6d6476b
olmocr/train/train.py
@@ -188,7 +188,7 @@ def main():
188
checkpoint_dirs.sort(key=lambda x: int(x.split("-")[1]))
189
latest_checkpoint = os.path.join(full_output_dir, checkpoint_dirs[-1])
190
logger.info(f"Found existing checkpoint: {latest_checkpoint}")
191
- found_resumable_checkpoint = True
+ found_resumable_checkpoint = latest_checkpoint
192
else:
193
logger.info("No existing checkpoints found in output directory")
194
0 commit comments