github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

rename shared adam to sharded optim v2

pull/394/head
ver217 2022-03-03 15:55:27 +08:00 committed by Frank Lee
parent 70814dc22f
commit b105371ace
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
colossalai/zero/sharded_optim/sharded_adam.py → colossalai/zero/sharded_optim/sharded_optim_v2.py
View File

@ -22,7 +22,7 @@ class OptimState(Enum):
UNSCALED = 2 UNSCALED = 2
class ShardedAdam(ColossalaiOptimizer): class ShardedOptimizerV2(ColossalaiOptimizer):
def __init__(self, def __init__(self,
adam_optim: Optimizer, adam_optim: Optimizer,