github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-07 03:52:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

[pipeline] refactor 1f1b schedule (#4115)

Browse Source 
* [api] update optimizer wrapper to fit pipeline

* [pipeline] add base schedule

* [pipeline] add 1f1b schedule

* [test] add pipeline schedule utils test

* [pipeline] fix import
This commit is contained in:
Hongxin Liu
2023-06-29 13:35:39 +08:00
parent 45fdc9b42c
commit f51ce1bc8e
6 changed files with 451 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
colossalai/pipeline/schedule/__init__.py Normal file
View File

@@ -0,0 +1,7 @@
from .base import PipelineSchedule
from .one_f_one_b import OneForwardOneBackwardSchedule
__all__ = [
'PipelineSchedule',
'OneForwardOneBackwardSchedule',
]