ver217 7bf1e98b97 pipeline last stage supports multi output (#151)

2022-01-17 15:57:47 +08:00

..

__init__.py

add interleaved pipeline, fix naive amp and update pipeline model initializer (#80)

2021-12-20 23:26:19 +08:00

_base_schedule.py

Added MoE parallel (#127)

2022-01-07 15:08:36 +08:00

_non_pipeline_schedule.py

Optimize pipeline schedule (#94)

2021-12-30 15:56:46 +08:00

_pipeline_schedule.py

pipeline last stage supports multi output (#151)

2022-01-17 15:57:47 +08:00