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"],
|
logging_steps=config["train_params"]["logging_steps"],
|
||||||
save_steps=config["train_params"]["save_steps"],
|
save_steps=config["train_params"]["save_steps"],
|
||||||
save_total_limit=config["train_params"]["save_total_limit"],
|
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),
|
eval_steps=config.get("eval_steps", 100),
|
||||||
mixed_precision=config.get("mixed_precision", "bf16"),
|
mixed_precision=config.get("mixed_precision", "bf16"),
|
||||||
gradient_checkpointing=config.get("gradient_checkpointing", True),
|
gradient_checkpointing=config.get("gradient_checkpointing", True),
|
||||||
|
|||||||
Reference in New Issue
Block a user