ColossalAI/colossalai/amp/naive_amp
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
_fp16_optimizer.py update examples and sphnix docs for the new api (#63) 2021-12-13 22:07:01 +08:00
naive_amp.py update examples and sphnix docs for the new api (#63) 2021-12-13 22:07:01 +08:00