fix: evaluation_strategy -> eval_strategy API change
This commit is contained in:
1 parent
50fa171f68
commit
9280103260
1 file changed
+1
-1
@@ -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