Revert "[megatron] fix: BF16 mode should use PAO as well" (#4234)

ISEEKYAN · web-flow · commit fc7df6f7f99b · 2025-11-22T13:17:06.000+08:00
Reverts #4221
diff --git a/verl/utils/megatron/optimizer.py b/verl/utils/megatron/optimizer.py
@@ -47,8 +47,6 @@ def init_megatron_optim_config(optim_config: dict, fp16: bool = False) -> Optimi
             {
                 "bf16": True,
                 "params_dtype": torch.bfloat16,
-                "use_precision_aware_optimizer": True,
-                "store_param_remainders": True,
             }
         )
     override_config = optim_config.get("override_optimizer_config", {})

Original file line number	Diff line number	Diff line change
`@@ -47,8 +47,6 @@ def init_megatron_optim_config(optim_config: dict, fp16: bool = False) -> Optimi`
`47`	`47`	`{`
`48`	`48`	`"bf16": True,`
`49`	`49`	`"params_dtype": torch.bfloat16,`
`50`		`- "use_precision_aware_optimizer": True,`
`51`		`- "store_param_remainders": True,`
`52`	`50`	`}`
`53`	`51`	`)`
`54`	`52`	`override_config = optim_config.get("override_optimizer_config", {})`