Commit 30410140 authored by Ziheng Qin's avatar Ziheng Qin Committed by Frank Lee
Browse files

[NFC] polish colossalai/amp/naive_amp/grad_scaler/dynamic_grad_scaler.py code style (#2299)


Co-authored-by: default avatarhenryqin1997 <henryqin1997@gamil.com>
parent 49715a78
#!/usr/bin/env python #!/usr/bin/env python
# -*- encoding: utf-8 -*- # -*- encoding: utf-8 -*-
from typing import Optional
import torch import torch
from .base_grad_scaler import BaseGradScaler from .base_grad_scaler import BaseGradScaler
from typing import Optional
__all__ = ['DynamicGradScaler'] __all__ = ['DynamicGradScaler']
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment