mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-05-16 12:20:45 +00:00
6e51d296f0
* add zero1&2 optimizer * rename test ditectory * rename test files * change tolerance in test
5 lines
159 B
Python
5 lines
159 B
Python
from .low_level_optim import LowLevelZeroOptimizer
|
|
from .sharded_optim_v2 import ShardedOptimizerV2
|
|
|
|
__all__ = ['ShardedOptimizerV2', 'LowLevelZeroOptimizer']
|