mirror of https://github.com/hpcaitech/ColossalAI
400f63012e
* add alpa dp split * add alpa dp split * use fwd+bwd instead of fwd only --------- Co-authored-by: Ziyue Jiang <ziyue.jiang@gmail.com> |
||
---|---|---|
.. | ||
__init__.py | ||
_pipeline_base.py | ||
_pipeline_schedule.py | ||
utils.py |