ColossalAI/colossalai/amp
Frank Lee a4e91bc87f
[bug] fixed grad scaler compatibility with torch 1.8 (#735)
2022-04-12 16:04:21 +08:00
..
apex_amp html refactor (#555) 2022-03-31 11:36:56 +08:00
naive_amp polish amp docstring (#616) 2022-04-01 16:09:39 +08:00
torch_amp [bug] fixed grad scaler compatibility with torch 1.8 (#735) 2022-04-12 16:04:21 +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