github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-07-14 15:44:46 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source 
Co-authored-by: RichardoLuo <14049555596@qq.com>
This commit is contained in:
RichardoLuo 2022-04-01 23:41:09 +08:00 committed by binmakeswell
parent 2e11853d04
commit ad1e7ab2b2
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
colossalai/engine/_base_engine.py
View File

@ -15,6 +15,7 @@ from typing import Optional, Type
from colossalai.engine.gradient_handler import BaseGradientHandler from colossalai.engine.gradient_handler import BaseGradientHandler
from colossalai.logging import get_dist_logger from colossalai.logging import get_dist_logger
class Engine: class Engine:
"""Basic engine class for training and evaluation. It runs a specific process method """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. :meth:`step` which is based on the given :attr:`schedule` over each batch of a dataset.