ColossalAI/colossalai/pipeline
Frank Lee 2b2dc1c86b
[pipeline] refactor the pipeline module (#1087)
* [pipeline] refactor the pipeline module

* polish code
2022-06-10 11:27:38 +08:00
..
__init__.py [pipeline] refactor the pipeline module (#1087) 2022-06-10 11:27:38 +08:00
layer_sepc.py [pipeline] refactor the pipeline module (#1087) 2022-06-10 11:27:38 +08:00
pipelinable.py [pipeline] refactor the pipeline module (#1087) 2022-06-10 11:27:38 +08:00
utils.py [pipeline] refactor the pipeline module (#1087) 2022-06-10 11:27:38 +08:00