github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-21 21:22:04 +00:00
Code Issues Packages Projects Releases Wiki Activity

rename shared adam to sharded optim v2

Browse Source
This commit is contained in:
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,