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.

Ziyue Jiang 400f63012e [pipeline] Add Simplified Alpa DP Partition (#2507) ... * add alpa dp split * add alpa dp split * use fwd+bwd instead of fwd only --------- Co-authored-by: Ziyue Jiang <ziyue.jiang@gmail.com>		2 years ago
..
__init__.py	[pipeline/pytree] add pytree to process args and kwargs | provide `data_process_func` to process args and kwargs after forward (#1642)	2 years ago
_pipeline_base.py	[pipeline] Add Simplified Alpa DP Partition (#2507)	2 years ago
_pipeline_schedule.py	polish pp middleware (#2476)	2 years ago
utils.py	fix type of num_worker_threads (#2237)	2 years ago