ColossalAI/colossalai/engine/schedule
YuliangLiu0306 32a45cd7ef
[pipelinable]use pipelinable to support GPT model. (#903)
* [CLI] add CLI launcher

* Revert "[CLI] add CLI launcher"

This reverts commit df7e6506d4.

* [pipelinable]use pipelinable to support GPT model.

* fix a bug caused by ShardedModel

* polish

* fix front func list
2022-05-11 09:23:58 +08:00
..
__init__.py [pipeline] refactor pipeline (#679) 2022-04-07 15:54:14 +08:00
_base_schedule.py 'fix/format' (#573) 2022-04-06 11:40:59 +08:00
_non_pipeline_schedule.py [refactor] pipeline, put runtime schedule into engine. (#627) 2022-04-03 20:46:45 +08:00
_pipeline_schedule.py [pipelinable]use pipelinable to support GPT model. (#903) 2022-05-11 09:23:58 +08:00