yujunjun/olmocr
1
0
Fork 0
mirror of https://github.com/allenai/olmocr.git synced 2025-11-04 03:56:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

2epoch config fix

Browse Source
This commit is contained in:
Jake Poznanski 2025-08-11 22:21:55 +00:00
parent 387e7947c4
commit 98d457c502
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
olmocr/train/configs/qwen25_vl_olmocrv3_rotation_2epoch.yaml
View File

@ -2,7 +2,7 @@
# Project metadata # Project metadata
project_name: olmocr-qwen-vl-training project_name: olmocr-qwen-vl-training
run_name: qwen2.5-vl-7b-olmocrv3_2epoch_prompt_first_rotation run_name: qwen2.5-vl-7b-olmocrv3_2epoch_prompt_first_rotation_fixed
# Model configuration # Model configuration
model: model:
@ -60,7 +60,7 @@ dataset:
# Training configuration # Training configuration
training: training:
output_dir: /weka/oe-data-default/jakep/olmocr-trainer/ output_dir: /weka/oe-data-default/jakep/olmocr-trainer/
num_train_epochs: 1 num_train_epochs: 2
# Batch size and accumulation # Batch size and accumulation
per_device_train_batch_size: 1 per_device_train_batch_size: 1