Options

2025-10-17 11:12:33 +00:00 · 2025-08-21 16:55:24 +00:00 · 2025-08-21 16:55:24 +00:00 · f865e49624
commit f865e49624
parent 3c8410f22c
1 changed files with 0 additions and 2 deletions
--- a/olmocr/train/grpo_train.py
+++ b/olmocr/train/grpo_train.py
@ -408,9 +408,7 @@ def main():
        temperature=0.7,
        report_to=report_to,
        remove_unused_columns=False,
-        torch_dtype=torch.bfloat16,
        bf16=True,
-        gradient_checkpointing=True,
        dataloader_num_workers=0,
    )