github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/amp/naive_amp/grad_scaler/__init__.py

6 lines
222 B
Python
Raw Normal View History

refactored grad scaler (#338)
2022-03-09 03:52:43 +00:00
from .base_grad_scaler import BaseGradScaler
from .constant_grad_scaler import ConstantGradScaler
from .dynamic_grad_scaler import DynamicGradScaler
[misc] update pre-commit and run all files (#4752) * [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format
2023-09-19 06:20:26 +00:00
__all__ = ["BaseGradScaler", "ConstantGradScaler", "DynamicGradScaler"]