github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-07 02:54:10 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
Co-authored-by: henryqin1997 <henryqin1997@gamil.com>
This commit is contained in:
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']