diff --git a/olmocr/train/configs/qwen25_vl_olmocrv2_1epoch.yaml b/olmocr/train/configs/qwen25_vl_olmocrv2_1epoch.yaml
index 246f58d..9b79e92 100644
--- a/olmocr/train/configs/qwen25_vl_olmocrv2_1epoch.yaml
+++ b/olmocr/train/configs/qwen25_vl_olmocrv2_1epoch.yaml
@@ -57,7 +57,7 @@ dataset:
 # Training configuration
 training:
   output_dir: /weka/oe-data-default/jakep/olmocr-trainer/
-  num_train_epochs: 1.0
+  num_train_epochs: 1
   
   # Batch size and accumulation
   per_device_train_batch_size: 1