github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-29 16:57:55 +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
parent df34bd0c7f
commit 5cc84d94dc
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
class ShardedAdam(ColossalaiOptimizer):
class ShardedOptimizerV2(ColossalaiOptimizer):
def __init__(self,
adam_optim: Optimizer,