Skip to content

GitLab

"tests/git@developer.sourcefind.cn:OpenDAS/colossalai.git" did not exist on "7cfed5f076f683293eb559c7e61feb6d3376b8d3"
Unverified Commit 104cbbb3 authored by ver217's avatar ver217 Committed by GitHub
Browse files

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

parent e6d50ec1
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
colossalai/nn/optimizer/__init__.py
View file @ 104cbbb3
......@@ -5,7 +5,6 @@ from .fused_sgd import FusedSGD
from .lamb import Lamb
from .lars import Lars
from .cpu_adam import CPUAdam
from .hybrid_adam import HybridAdam
__all__ = [
'ColossalaiOptimizer', 'FusedLAMB', 'FusedAdam', 'FusedSGD', 'Lamb', 'Lars', 'CPUAdam'
]
__all__ = ['ColossalaiOptimizer', 'FusedLAMB', 'FusedAdam', 'FusedSGD', 'Lamb', 'Lars', 'CPUAdam', 'HybridAdam']
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment