mirror of
https://github.com/allenai/olmocr.git
synced 2025-11-05 04:24:09 +00:00
2epoch config fix
This commit is contained in:
parent
387e7947c4
commit
98d457c502
@ -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
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user