Commit ad1e7ab2 authored by RichardoLuo's avatar RichardoLuo Committed by binmakeswell
Browse files

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


Co-authored-by: default avatarRichardoLuo <14049555596@qq.com>
parent 2e11853d
......@@ -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.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment