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.

Hongxin Liu 1b387ca9fe [shardformer] refactor pipeline grad ckpt config (#5646) ... * [shardformer] refactor pipeline grad ckpt config * [shardformer] refactor pipeline grad ckpt config * [pipeline] fix stage manager		7 months ago
..
schedule	[shardformer] fix chatglm implementation (#5644)	7 months ago
__init__.py	[misc] update pre-commit and run all files (#4752)	1 year ago
p2p.py	[pipeline] A more general _communicate in p2p (#5062)	11 months ago
stage_manager.py	[shardformer] refactor pipeline grad ckpt config (#5646)	7 months ago