rename shared adam to sharded optim v2

b105371a · ver217 · Frank Lee · 70814dc2 · b105371a
Commit b105371a authored Mar 03, 2022 by ver217 Committed by Frank Lee Mar 11, 2022
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

colossalai/zero/sharded_optim/sharded_optim_v2.py colossalai/zero/sharded_optim/sharded_optim_v2.py +1 -1

No files found.
--- a/colossalai/zero/sharded_optim/sharded_adam.py
+++ b/colossalai/zero/sharded_optim/sharded_adam.py
@@ -22,7 +22,7 @@ class OptimState(Enum):
    UNSCALED = 2
-class ShardedAdam(ColossalaiOptimizer):
+class ShardedOptimizerV2(ColossalaiOptimizer):
    def __init__(self,
                 adam_optim: Optimizer,