[CPUAdam] Update full_precision_optimizer_states in docstring (#5181)

It appears this variable was renamed to `fp32_optimizer_states` so reflecting this change in docstrings. Co-authored-by: Olatunji Ruwase <[email protected]>
microsoft · Feb 26, 2024 · 8c42a30 · 8c42a30
1 parent 4874941
commit 8c42a30
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deepspeed/ops/adam/cpu_adam.py b/deepspeed/ops/adam/cpu_adam.py
@@ -63,7 +63,7 @@ def __init__(self,
                 algorithm from the paper `On the Convergence of Adam and Beyond`_
                 (default: False) NOT SUPPORTED in DeepSpeed CPUAdam!
             adamw_mode: select between Adam and AdamW implementations (default: AdamW)
-            full_precision_optimizer_states: creates momentum and variance in full precision regardless of
+            fp32_optimizer_states: creates momentum and variance in full precision regardless of
                         the precision of the parameters (default: True)
         """