ColossalAI/colossalai/zero
Frank Lee 35813ed3c4
update examples and sphnix docs for the new api (#63)
2021-12-13 22:07:01 +08:00
..
__init__.py update examples and sphnix docs for the new api (#63) 2021-12-13 22:07:01 +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 fix zero3 fp16 and add zero3 model context (#62) 2021-12-10 17:48:50 +08:00