update train hparams

This commit is contained in:
hiyouga committed 2024-06-06 01:49:20 +08:00
1 parent 4dc0632145
commit dc4a00dd63
22 files changed
+22 -22

No files matched your search

+1 -1
View File
@@ -200,7 +200,7 @@ class Runner:
# eval config
if get("train.val_size") > 1e-6 and args["stage"] != "ppo":
args["val_size"] = get("train.val_size")
args["evaluation_strategy"] = "steps"
args["eval_strategy"] = "steps"
args["eval_steps"] = args["save_steps"]
args["per_device_eval_batch_size"] = args["per_device_train_batch_size"]