'[NFC] polish <colossalai/engine/_base_engine.py> code style' (#631)

Co-authored-by: RichardoLuo <14049555596@qq.com>
2022-04-01 23:41:09 +08:00 · 2022-04-01 23:41:09 +08:00 · ad1e7ab2b2
parent 2e11853d04
commit ad1e7ab2b2
1 changed files with 1 additions and 0 deletions
--- 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.