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

[hotfix] add hybrid adam to __init__ (#584)

pull/577/head^2
ver217 3 years ago committed by GitHub
parent
e6d50ec107
commit
104cbbb313
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 3 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 5
      colossalai/nn/optimizer/__init__.py

5
colossalai/nn/optimizer/__init__.py
Unescape View File

@ -5,7 +5,6 @@ from .fused_sgd import FusedSGD
from .lamb import Lamb from .lamb import Lamb
from .lars import Lars from .lars import Lars
from .cpu_adam import CPUAdam from .cpu_adam import CPUAdam
from .hybrid_adam import HybridAdam
__all__ = [ __all__ = ['ColossalaiOptimizer', 'FusedLAMB', 'FusedAdam', 'FusedSGD', 'Lamb', 'Lars', 'CPUAdam', 'HybridAdam']
'ColossalaiOptimizer', 'FusedLAMB', 'FusedAdam', 'FusedSGD', 'Lamb', 'Lars', 'CPUAdam'
]

Write Preview
Loading…
Cancel
Save