ColossalAI/naive_amp at feature/async-io - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-13 05:01:44 +00:00

Files

History

Hongxin Liu a15ab139ad [plugin] support get_grad_norm (#6115 )

2024-11-05 18:12:47 +08:00

..

[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