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

Checking to see if specifying number of dataloader workers helps thread issue

Browse Source
This commit is contained in:
Jake Poznanski 2025-09-22 17:08:06 +00:00
parent 3b20322eb0
commit 1a64728420
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/grpo_train.py
View File

@ -1047,7 +1047,7 @@ def main():
report_to=report_to, report_to=report_to,
remove_unused_columns=False, remove_unused_columns=False,
bf16=True, bf16=True,
dataloader_num_workers=0, dataloader_num_workers=8,
dataloader_drop_last=True, dataloader_drop_last=True,
# GRPO-specific parameters # GRPO-specific parameters