yujunjun/olmocr
1
0
Fork 0
mirror of https://github.com/allenai/olmocr.git synced 2025-11-13 17:08:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix

Browse Source
This commit is contained in:
Jake Poznanski 2024-10-16 13:30:06 -07:00
parent d94713e73e
commit 51f1669451
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
pdelfin/train/train.py
View File

@ -45,7 +45,6 @@ from .utils import (
RunName, RunName,
get_local_dir, get_local_dir,
log_trainable_parameters, log_trainable_parameters,
packing_collator,
setup_environment, setup_environment,
make_dataset, make_dataset,
TruncatingCollator TruncatingCollator
@ -169,7 +168,7 @@ def run_train(config: TrainConfig):
label_names=["labels"], # fix from https://github.com/huggingface/transformers/issues/22885 label_names=["labels"], # fix from https://github.com/huggingface/transformers/issues/22885
max_grad_norm=config.hparams.clip_grad_norm, max_grad_norm=config.hparams.clip_grad_norm,
remove_unused_columns=False, remove_unused_columns=False,
#eval_on_start=True, eval_on_start=True,
metric_for_best_model=config.valid_data.metric_for_best_model, metric_for_best_model=config.valid_data.metric_for_best_model,
) )