github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

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

Co-authored-by: henryqin1997 <henryqin1997@gamil.com>
pull/2317/head
Ziheng Qin 2023-01-04 10:11:54 +08:00 committed by Frank Lee
parent 49715a78f0
commit 3041014089
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
colossalai/amp/naive_amp/grad_scaler/dynamic_grad_scaler.py
View File

@ -1,10 +1,12 @@
#!/usr/bin/env python
# -*- encoding: utf-8 -*-
import torch
from .base_grad_scaler import BaseGradScaler
from typing import Optional
import torch
from .base_grad_scaler import BaseGradScaler
__all__ = ['DynamicGradScaler']