fix: evaluation_strategy -> eval_strategy API change

This commit is contained in:
Christian Medina
2026-06-30 20:29:03 -04:00
parent 50fa171f68
commit 9280103260

View File

@@ -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),