From 38517176ea9fe6eb1fb16a4dc01a3257434a7492 Mon Sep 17 00:00:00 2001 From: Jake Poznanski Date: Tue, 22 Jul 2025 18:13:38 +0000 Subject: [PATCH] Fix --- olmocr/train/configs/qwen25_vl_olmocrv2_1epoch.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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