This commit is contained in:
Jake Poznanski 2025-07-22 18:13:38 +00:00
parent 731ca95fda
commit 38517176ea

View File

@ -57,7 +57,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.0 num_train_epochs: 1
# Batch size and accumulation # Batch size and accumulation
per_device_train_batch_size: 1 per_device_train_batch_size: 1