Skip to content

Commit

Permalink
'[NFC] polish <colossalai/engine/_base_engine.py> code style' (hpcait…
Browse files Browse the repository at this point in the history
…ech#631)

Co-authored-by: RichardoLuo <14049555596@qq.com>
  • Loading branch information
2 people authored and binmakeswell committed Apr 6, 2022
1 parent 2e11853 commit ad1e7ab
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions colossalai/engine/_base_engine.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,6 +15,7 @@
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.
Expand Down

0 comments on commit ad1e7ab

Please sign in to comment.