ColossalAI/colossalai/amp
ver217 8f02a88db2
add interleaved pipeline, fix naive amp and update pipeline model initializer (#80)
2021-12-20 23:26:19 +08:00
..
apex_amp fixed zero level 3 dtype bug (#76) 2021-12-20 17:00:53 +08:00
naive_amp add interleaved pipeline, fix naive amp and update pipeline model initializer (#80) 2021-12-20 23:26:19 +08:00
torch_amp 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
amp_type.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00