set recompute_loss_function to false by default (#1089)

zhuzilin · web-flow · commit 6d43f5307cbd · 2025-12-11T17:05:54.000+08:00
diff --git a/slime/utils/arguments.py b/slime/utils/arguments.py
@@ -149,9 +149,8 @@ def add_train_arguments(parser):
                 help="The method to convert megatron weights to hugging face weights for SGLang.",
             )
             parser.add_argument(
-                "--disable-recompute-loss-function",
-                action="store_false",
-                dest="recompute_loss_function",
+                "--recompute-loss-function",
+                action="store_true",
                 help="Whether to disable recompute loss function to save memory during training.",
             )
 

Original file line number	Diff line number	Diff line change
`@@ -149,9 +149,8 @@ def add_train_arguments(parser):`
`149`	`149`	`help="The method to convert megatron weights to hugging face weights for SGLang.",`
`150`	`150`	`)`
`151`	`151`	`parser.add_argument(`
`152`		`- "--disable-recompute-loss-function",`
`153`		`- action="store_false",`
`154`		`- dest="recompute_loss_function",`
	`152`	`+ "--recompute-loss-function",`
	`153`	`+ action="store_true",`
`155`	`154`	`help="Whether to disable recompute loss function to save memory during training.",`
`156`	`155`	`)`
`157`	`156`