mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2026-04-25 01:03:35 +00:00
5 lines
147 B
Python
5 lines
147 B
Python
from .sharded_optim import ShardedOptimizer
|
|
from .sharded_optim_v2 import ShardedOptimizerV2
|
|
|
|
__all__ = ['ShardedOptimizer', 'ShardedOptimizerV2']
|