423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-12-14 19:06:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

update train hparams

Browse Source 
Former-commit-id: dc4a00dd63
This commit is contained in:
hiyouga
2024-06-06 01:49:20 +08:00
parent 8f3b8ade45
commit 00b3fb4d14
22 changed files with 22 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/llamafactory/extras/env.py
View File

@@ -51,4 +51,4 @@ def print_env() -> None:
info["vLLM version"] = vllm.__version__
print("\n".join(["- {}: {}".format(key, value) for key, value in info.items()]) + "\n")
print("\n" + "\n".join(["- {}: {}".format(key, value) for key, value in info.items()]) + "\n")

2
src/llamafactory/webui/runner.py
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"]