ColossalAI/colossalai/amp/torch_amp at 237d08e7eee24cb396914df660cc40d6dceaf4e4 - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-24 22:35:06 +00:00

Files

History

Jiarui Fang 496cbb0760 [hotfix] fix initialize bug with zero (#442 )

2022-03-17 13:16:22 +08:00

..

__init__.py

[hotfix] fix initialize bug with zero (#442 )

2022-03-17 13:16:22 +08:00

_grad_scaler.py

Optimize pipeline schedule (#94 )

2021-12-30 15:56:46 +08:00

torch_amp.py

Fixed docstring in colossalai (#171 )

2022-01-21 10:44:30 +08:00