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/_utils.py code style (#1816) 

* [NFC] polish colossalai/nn/metric/accuracy_2p5d.py code style (#1714)

* [NFC] polish colossalai/zero/sharded_param/__init__.py code style

* [NFC] polish colossalai/amp/naive_amp/_utils.py code style

Co-authored-by: shenggan <csg19971016@gmail.com>
Co-authored-by: ric <mkkt_bkkt@mail.ustc.edu.cn>
pull/1849/head
CsRic 2022-11-08 14:57:18 +08:00 committed by binmakeswell
parent 25993db98a
commit 9623ec1b02
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
colossalai/amp/naive_amp/_utils.py
View File

@ -1,4 +1,5 @@
from typing import List from typing import List
from torch import Tensor from torch import Tensor