yujunjun/olmocr
1
0
Fork 0
mirror of https://github.com/allenai/olmocr.git synced 2025-10-13 09:12:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

One idea for resume fix

Browse Source
This commit is contained in:
Jake Poznanski 2025-07-02 01:33:37 +00:00
parent 2a20607d37
commit 6d6476b31a
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
olmocr/train/train.py
View File

@ -188,7 +188,7 @@ def main():
checkpoint_dirs.sort(key=lambda x: int(x.split("-")[1])) checkpoint_dirs.sort(key=lambda x: int(x.split("-")[1]))
latest_checkpoint = os.path.join(full_output_dir, checkpoint_dirs[-1]) latest_checkpoint = os.path.join(full_output_dir, checkpoint_dirs[-1])
logger.info(f"Found existing checkpoint: {latest_checkpoint}") logger.info(f"Found existing checkpoint: {latest_checkpoint}")
found_resumable_checkpoint = True found_resumable_checkpoint = latest_checkpoint
else: else:
logger.info("No existing checkpoints found in output directory") logger.info("No existing checkpoints found in output directory")