423A35C7/LLaMA-Factory
1
0
Fork 0
mirror of https://github.com/hiyouga/LLaMA-Factory.git synced 2025-10-15 08:08:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

Update finetuning_args.py

Browse Source 
Former-commit-id: 30e3430553f1f7e09cd57ef2c9843b549746c618
This commit is contained in:
Yuchen Han 2023-11-17 00:15:51 -08:00 committed by GitHub
parent dc0f81aabc
commit bcd31cf245
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/llmtuner/hparams/finetuning_args.py
View File

@ -74,6 +74,10 @@ class RLHFArguments:
default=None,
metadata={"help": "Log with either 'wandb' or 'tensorboard' in PPO training."}
)
ppo_epochs: Optional[int] = field(
default=4,
metadata={"help": "Number of optimisation epochs per batch of samples"},
)
ppo_score_norm: Optional[bool] = field(
default=False,
metadata={"help": "Use score normalization in PPO training."}