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"],
|
||||
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