From cb885d38f28d2260e6bc9444cba846408a792325 Mon Sep 17 00:00:00 2001 From: Sean Lee Date: Mon, 30 Sep 2024 10:04:45 +0800 Subject: [PATCH] set evaluation_strategy no when valid_ds is empty --- angle_emb/angle.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/angle_emb/angle.py b/angle_emb/angle.py index b687ed8..77ff58e 100644 --- a/angle_emb/angle.py +++ b/angle_emb/angle.py @@ -1543,8 +1543,7 @@ def fit(self, logging_steps=logging_steps, save_steps=save_steps, save_strategy=save_strategy, - do_eval=valid_ds is not None, - evaluation_strategy=evaluation_strategy, + evaluation_strategy=evaluation_strategy if valid_ds is not None else 'no', eval_steps=eval_steps, output_dir=output_dir, save_total_limit=save_total_limit,