set evaluation_strategy no when valid_ds is empty

SeanLee97 · Sep 30, 2024 · cb885d3 · cb885d3
1 parent a9cc91e
commit cb885d3
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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,