ColossalAI

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

History

Frank Lee ed19290560 [booster] implemented mixed precision class (#3151 ) * [booster] implemented mixed precision class * polish code		2023-03-17 11:00:15 +08:00
..
interface	[booster] implemented mixed precision class (#3151 )	2023-03-17 11:00:15 +08:00
mixed_precision	[booster] implemented mixed precision class (#3151 )	2023-03-17 11:00:15 +08:00
__init__.py	[booster] implemented mixed precision class (#3151 )	2023-03-17 11:00:15 +08:00
accelerator.py	…
booster.py	[booster] implemented mixed precision class (#3151 )	2023-03-17 11:00:15 +08:00
environment_table.py	…
plugin.py	…