ColossalAI/colossalai/zero
ver217 9ef05ed1fc
try import deepspeed when using zero (#130)
2022-01-07 17:24:57 +08:00
..
__init__.py try import deepspeed when using zero (#130) 2022-01-07 17:24:57 +08:00
loss_scaler.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
zero_redundancy_optimizer_level_2.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
zero_redundancy_optimizer_level_3.py fixed zero level 3 dtype bug (#76) 2021-12-20 17:00:53 +08:00