ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI

History

HELSON 4c4388c46e [hotfix] fix memory leak in zero (#781 )		2022-04-18 13:57:03 +08:00
..
grad_scaler	[hotfix] fix memory leak in zero (#781 )	2022-04-18 13:57:03 +08:00
__init__.py	polish amp docstring (#616 )	2022-04-01 16:09:39 +08:00
_fp16_optimizer.py	…
_utils.py	…
naive_amp.py	…