ColossalAI/colossalai/amp
Frank Lee 72bd7c696b
[amp] included dict for type casting of model output (#1102)
2022-06-13 14:18:04 +08:00
..
apex_amp [doc] improved docstring in the amp module (#857) 2022-04-25 13:42:17 +08:00
naive_amp [amp] included dict for type casting of model output (#1102) 2022-06-13 14:18:04 +08:00
torch_amp [doc] improved docstring in the amp module (#857) 2022-04-25 13:42:17 +08:00
__init__.py [refactor] zero directory (#724) 2022-04-11 23:13:02 +08:00
amp_type.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00