From 9623ec1b02d345b3515ad5b666a9ebfa79dbe69e Mon Sep 17 00:00:00 2001 From: CsRic <59389055+CsRic@users.noreply.github.com> Date: Tue, 8 Nov 2022 14:57:18 +0800 Subject: [PATCH] [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 Co-authored-by: ric --- colossalai/amp/naive_amp/_utils.py | 1 + 1 file changed, 1 insertion(+) diff --git a/colossalai/amp/naive_amp/_utils.py b/colossalai/amp/naive_amp/_utils.py index ad2a2ceed..7633705e1 100644 --- a/colossalai/amp/naive_amp/_utils.py +++ b/colossalai/amp/naive_amp/_utils.py @@ -1,4 +1,5 @@ from typing import List + from torch import Tensor