support ppo score norm (trl 0.5.1.dev required)

This commit is contained in:
hiyouga committed 2023-08-18 12:02:42 +08:00
1 parent 9020524418
commit 53e33418d0
2 files changed
+10

No files matched your search

+4
View File
@@ -61,6 +61,10 @@ class FinetuningArguments:
default=True,
metadata={"help": "Whether to resume training from the last LoRA weights or create new weights after merging them."}
)
ppo_score_norm: Optional[bool] = field(
default=False,
metadata={"help": "Use score normalization in PPO Training."}
)
dpo_beta: Optional[float] = field(
default=0.1,
metadata={"help": "The beta parameter for the DPO loss."}