From ad1e7ab2b215bff7e3973562cac2f3bb31454ba1 Mon Sep 17 00:00:00 2001 From: RichardoLuo <50363844+RichardoLuo@users.noreply.github.com> Date: Fri, 1 Apr 2022 23:41:09 +0800 Subject: [PATCH] '[NFC] polish code style' (#631) Co-authored-by: RichardoLuo <14049555596@qq.com> --- colossalai/engine/_base_engine.py | 1 + 1 file changed, 1 insertion(+) diff --git a/colossalai/engine/_base_engine.py b/colossalai/engine/_base_engine.py index 4f3654cd9..696d82e41 100644 --- a/colossalai/engine/_base_engine.py +++ b/colossalai/engine/_base_engine.py @@ -15,6 +15,7 @@ from typing import Optional, Type from colossalai.engine.gradient_handler import BaseGradientHandler from colossalai.logging import get_dist_logger + class Engine: """Basic engine class for training and evaluation. It runs a specific process method :meth:`step` which is based on the given :attr:`schedule` over each batch of a dataset.