You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

Jianghai 90a65ea682 [pipeline] build bloom model and policy , revise the base class of policy (#4161) ... * add pipeline policy and bert forward to be done * add bertmodel pipeline forward and make tests * add Bert_Policy and test for policy * update formatting * update formatting * update the code * fix bugs * fix name confilt * add bloom model and policy ,revise the base class of policy * revise * revision * add bert_for_pretraining		1 year ago
..
middleware	[Pipeline Middleware] Adapt scheduler for Topo (#2066)	2 years ago
policy	[pipeline] build bloom model and policy , revise the base class of policy (#4161)	1 year ago
rpc	[nfc]fix typo colossalai/pipeline tensor nn (#3899)	2 years ago
schedule	[pipeline] refactor 1f1b schedule (#4115)	1 year ago
__init__.py	fix file name (#1759)	2 years ago
layer_spec.py	fix file name (#1759)	2 years ago
p2p.py	[pipeline] implement p2p communication (#4100)	1 year ago
pipelinable.py	[nfc]fix typo colossalai/pipeline tensor nn (#3899)	2 years ago
pipeline_process_group.py	[pipeline/fix-bug] num_microbatches support any integrate | stable chimera | launch tool for rpc pp framework (#1684)	2 years ago
stage_manager.py	[pipeline] add stage manager (#4093)	1 year ago
utils.py	[nfc]fix typo colossalai/pipeline tensor nn (#3899)	2 years ago