fix bugs

ShuoXing98 · ShuoXing98 · commit a82ce56b3393 · 2025-04-21T16:39:08.000-05:00
diff --git a/train_rdpo.py b/train_rdpo.py
@@ -157,6 +157,11 @@ class TrainingArguments(transformers.TrainingArguments):
         metadata={"help": "todo"}
     )
 
+    sft_weight: float = field(
+        default=0.0,
+        metadata={"help": "todo"}
+    )
+
 
 
 def maybe_zero_3(param, ignore_status=False, name=None):

Original file line number	Diff line number	Diff line change
`@@ -157,6 +157,11 @@ class TrainingArguments(transformers.TrainingArguments):`
`157`	`157`	`metadata={"help": "todo"}`
`158`	`158`	`)`
`159`	`159`
	`160`	`+ sft_weight: float = field(`
	`161`	`+ default=0.0,`
	`162`	`+ metadata={"help": "todo"}`
	`163`	`+ )`
	`164`	`+`
`160`	`165`
`161`	`166`
`162`	`167`	`def maybe_zero_3(param, ignore_status=False, name=None):`