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.
ColossalAI/tests/test_pipeline/test_pipeline_utils
Hongxin Liu 1b387ca9fe
[shardformer] refactor pipeline grad ckpt config (#5646)
7 months ago
..
test_t5_pipeline_utils.py [shardformer] refactor pipeline grad ckpt config (#5646) 7 months ago
test_whisper_pipeline_utils.py [shardformer] refactor pipeline grad ckpt config (#5646) 7 months ago