rename shared adam to sharded optim v2

2022-03-03 15:55:27 +08:00 · 2022-03-03 15:55:27 +08:00 · b105371ace
parent 70814dc22f
commit b105371ace
1 changed files with 1 additions and 1 deletions
--- a/colossalai/zero/sharded_optim/sharded_optim_v2.py
+++ b/colossalai/zero/sharded_optim/sharded_optim_v2.py
@ -22,7 +22,7 @@ class OptimState(Enum):
    UNSCALED = 2


-class ShardedAdam(ColossalaiOptimizer):
+class ShardedOptimizerV2(ColossalaiOptimizer):

    def __init__(self,
                 adam_optim: Optimizer,