From 51f16694511f678906e6c80fac3549a7ff8db900 Mon Sep 17 00:00:00 2001 From: Jake Poznanski Date: Wed, 16 Oct 2024 13:30:06 -0700 Subject: [PATCH] fix --- pdelfin/train/train.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/pdelfin/train/train.py b/pdelfin/train/train.py index 90f3e96..1f6fd90 100644 --- a/pdelfin/train/train.py +++ b/pdelfin/train/train.py @@ -45,7 +45,6 @@ RunName, get_local_dir, log_trainable_parameters, - packing_collator, setup_environment, make_dataset, TruncatingCollator @@ -169,7 +168,7 @@ def run_train(config: TrainConfig): label_names=["labels"], # fix from https://github.com/huggingface/transformers/issues/22885 max_grad_norm=config.hparams.clip_grad_norm, remove_unused_columns=False, - #eval_on_start=True, + eval_on_start=True, metric_for_best_model=config.valid_data.metric_for_best_model, )