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

Logs

Browse Source
This commit is contained in:
Jake Poznanski 2025-08-21 17:11:33 +00:00
parent 6fe630516a
commit de719edf49
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
olmocr/train/grpo_train.py
View File

@ -170,9 +170,10 @@ class OlmOCRDataset(Dataset):
# Return None if processing fails # Return None if processing fails
return None return None
def simple_length_reward(completions_ids, **kwargs): def simple_length_reward(completion_ids, **kwargs):
"""Reward function that assigns higher scores to longer completions (in terms of token count).""" """Reward function that assigns higher scores to longer completions (in terms of token count)."""
return [float(len(ids)) for ids in completions_ids] logger.info(f"Reward function called {kwargs}")
return [float(len(ids)) for ids in completion_ids]
def main(): def main():