fix: evaluation_strategy -> eval_strategy API change
This commit is contained in:
@@ -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),
|
||||
|
||||
Reference in New Issue
Block a user