From c5eb3fba621d2432a50b300a493ab5a97574ce28 Mon Sep 17 00:00:00 2001 From: hiyouga Date: Sun, 25 Feb 2024 20:23:41 +0800 Subject: [PATCH] update webui Former-commit-id: 298a5fc52610deb9f7d555e2fc699f10067d8af5 --- src/llmtuner/webui/runner.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/llmtuner/webui/runner.py b/src/llmtuner/webui/runner.py index 66465057..644e6495 100644 --- a/src/llmtuner/webui/runner.py +++ b/src/llmtuner/webui/runner.py @@ -165,7 +165,8 @@ class Runner: if get("train.val_size") > 1e-6 and args["stage"] != "ppo": args["val_size"] = get("train.val_size") args["evaluation_strategy"] = "steps" - args["eval_steps"] = get("train.save_steps") + args["eval_steps"] = args["save_steps"] + args["per_device_eval_batch_size"] = args["per_device_train_batch_size"] args["load_best_model_at_end"] = args["stage"] not in ["rm", "ppo"] return args