diff --git a/training/scripts/train.py b/training/scripts/train.py index cb6b7bf..b99aaee 100755 --- a/training/scripts/train.py +++ b/training/scripts/train.py @@ -87,7 +87,7 @@ def train(config_path): logging_steps=config["train_params"]["logging_steps"], save_steps=config["train_params"]["save_steps"], save_total_limit=config["train_params"]["save_total_limit"], - evaluation_strategy=config.get("eval_strategy", "steps"), + eval_strategy=config.get("eval_strategy", "steps"), eval_steps=config.get("eval_steps", 100), mixed_precision=config.get("mixed_precision", "bf16"), gradient_checkpointing=config.get("gradient_checkpointing", True),