ColossalAI/colossalai/pipeline/schedule
Baizhou Zhang 0387a47e63
[shardformer] fix emerged bugs after updating transformers (#4526)
2023-08-29 11:25:05 +08:00
..
__init__.py
_utils.py [shardformer] fix emerged bugs after updating transformers (#4526) 2023-08-29 11:25:05 +08:00
base.py
interleaved_pp.py [shardformer] support interleaved pipeline (#4448) 2023-08-16 19:29:03 +08:00
one_f_one_b.py [shardformer] zero1+pp and the corresponding tests (#4517) 2023-08-28 10:51:16 +08:00