github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-06-26 07:22:12 +00:00
Code Issues Packages Projects Releases Wiki Activity
feature/async-io
ColossalAI/colossalai/amp/naive_amp
History
Hongxin Liu a15ab139ad
[plugin] support get_grad_norm (#6115)
2024-11-05 18:12:47 +08:00
..
grad_scaler [npu] change device to accelerator api (#5239) 2024-01-09 10:20:05 +08:00
mixed_precision_mixin [npu] change device to accelerator api (#5239) 2024-01-09 10:20:05 +08:00
__init__.py [legacy] clean up legacy code (#4743) 2023-09-18 16:31:06 +08:00
mixed_precision_optimizer.py [plugin] support get_grad_norm (#6115) 2024-11-05 18:12:47 +08:00