update train hparams

This commit is contained in:
hiyouga 2024-06-06 01:49:20 +08:00
parent 4dc0632145
commit dc4a00dd63
22 changed files with 22 additions and 22 deletions

View File

@ -37,5 +37,5 @@ pure_bf16: true
### eval
val_size: 0.1
per_device_eval_batch_size: 1
evaluation_strategy: steps
eval_strategy: steps
eval_steps: 500

View File

@ -38,5 +38,5 @@ fp16: true
### eval
val_size: 0.1
per_device_eval_batch_size: 1
evaluation_strategy: steps
eval_strategy: steps
eval_steps: 500

View File

@ -38,5 +38,5 @@ pure_bf16: true
### eval
val_size: 0.1
per_device_eval_batch_size: 1
evaluation_strategy: steps
eval_strategy: steps
eval_steps: 500

View File

@ -36,5 +36,5 @@ fp16: true
### eval
val_size: 0.1
per_device_eval_batch_size: 1
evaluation_strategy: steps
eval_strategy: steps
eval_steps: 500

View File

@ -35,5 +35,5 @@ fp16: true
### eval
val_size: 0.1
per_device_eval_batch_size: 1
evaluation_strategy: steps
eval_strategy: steps
eval_steps: 500

View File

@ -35,5 +35,5 @@ pure_bf16: true
### eval
val_size: 0.1
per_device_eval_batch_size: 1
evaluation_strategy: steps
eval_strategy: steps
eval_steps: 500

View File

@ -37,5 +37,5 @@ fp16: true
### eval
val_size: 0.1
per_device_eval_batch_size: 1
evaluation_strategy: steps
eval_strategy: steps
eval_steps: 500

View File

@ -37,5 +37,5 @@ fp16: true
### eval
val_size: 0.1
per_device_eval_batch_size: 1
evaluation_strategy: steps
eval_strategy: steps
eval_steps: 500

View File

@ -38,5 +38,5 @@ fp16: true
### eval
val_size: 0.1
per_device_eval_batch_size: 1
evaluation_strategy: steps
eval_strategy: steps
eval_steps: 500

View File

@ -38,5 +38,5 @@ fp16: true
### eval
val_size: 0.1
per_device_eval_batch_size: 1
evaluation_strategy: steps
eval_strategy: steps
eval_steps: 500

View File

@ -36,5 +36,5 @@ fp16: true
### eval
val_size: 0.1
per_device_eval_batch_size: 1
evaluation_strategy: steps
eval_strategy: steps
eval_steps: 500

View File

@ -34,5 +34,5 @@ fp16: true
### eval
val_size: 0.1
per_device_eval_batch_size: 1
evaluation_strategy: steps
eval_strategy: steps
eval_steps: 500

View File

@ -33,5 +33,5 @@ fp16: true
### eval
val_size: 0.1
per_device_eval_batch_size: 1
evaluation_strategy: steps
eval_strategy: steps
eval_steps: 500

View File

@ -34,5 +34,5 @@ fp16: true
### eval
val_size: 0.1
per_device_eval_batch_size: 1
evaluation_strategy: steps
eval_strategy: steps
eval_steps: 500

View File

@ -34,5 +34,5 @@ fp16: true
### eval
val_size: 0.1
per_device_eval_batch_size: 1
evaluation_strategy: steps
eval_strategy: steps
eval_steps: 500

View File

@ -35,5 +35,5 @@ fp16: true
### eval
val_size: 0.1
per_device_eval_batch_size: 1
evaluation_strategy: steps
eval_strategy: steps
eval_steps: 500

View File

@ -34,5 +34,5 @@ fp16: true
### eval
val_size: 0.1
per_device_eval_batch_size: 1
evaluation_strategy: steps
eval_strategy: steps
eval_steps: 500

View File

@ -34,5 +34,5 @@ fp16: true
### eval
val_size: 0.1
per_device_eval_batch_size: 1
evaluation_strategy: steps
eval_strategy: steps
eval_steps: 500

View File

@ -35,5 +35,5 @@ fp16: true
### eval
val_size: 0.1
per_device_eval_batch_size: 1
evaluation_strategy: steps
eval_strategy: steps
eval_steps: 500

View File

@ -34,5 +34,5 @@ fp16: true
### eval
val_size: 0.1
per_device_eval_batch_size: 1
evaluation_strategy: steps
eval_strategy: steps
eval_steps: 500

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")

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