Former-commit-id: f513e1415cc3fe87f600318fba855d1286b6d007
This commit is contained in:
hiyouga 2024-03-26 17:30:12 +08:00
parent 3336422760
commit a916688723

View File

@ -102,7 +102,7 @@ class RLHFArguments:
default="sigmoid",
metadata={"help": "The type of DPO loss to use."},
)
dpo_label_smoothing = field(
dpo_label_smoothing: float = field(
default=0.0,
metadata={"help": "The robust DPO label smoothing parameter in cDPO that should be between 0 and 0.5."},
)