From f865e496242136034e3ae6d0e19b2a7fe3ef3fd3 Mon Sep 17 00:00:00 2001
From: Jake Poznanski <jakep@allenai.org>
Date: Thu, 21 Aug 2025 16:55:24 +0000
Subject: [PATCH] Options

---
 olmocr/train/grpo_train.py | 2 --
 1 file changed, 2 deletions(-)

diff --git a/olmocr/train/grpo_train.py b/olmocr/train/grpo_train.py
index 61859c0..b0c4822 100644
--- 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,
     )