From 3041014089ea7d3fe54a68c837a3efecaa1a080a Mon Sep 17 00:00:00 2001 From: Ziheng Qin <37519855+henryqin1997@users.noreply.github.com> Date: Wed, 4 Jan 2023 10:11:54 +0800 Subject: [PATCH] [NFC] polish colossalai/amp/naive_amp/grad_scaler/dynamic_grad_scaler.py code style (#2299) Co-authored-by: henryqin1997 --- colossalai/amp/naive_amp/grad_scaler/dynamic_grad_scaler.py | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/colossalai/amp/naive_amp/grad_scaler/dynamic_grad_scaler.py b/colossalai/amp/naive_amp/grad_scaler/dynamic_grad_scaler.py index 1ac26ee91..6d6f2f287 100644 --- a/colossalai/amp/naive_amp/grad_scaler/dynamic_grad_scaler.py +++ b/colossalai/amp/naive_amp/grad_scaler/dynamic_grad_scaler.py @@ -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']