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_shardformer
Hongxin Liu bd18678478
[test] fix gemini checkpoint and gpt test (#4620)
1 year ago
..
test_layer [shardformer] Add overlap support for gpt2 (#4535) 1 year ago
test_model [test] fix gemini checkpoint and gpt test (#4620) 1 year ago
__init__.py [shardformer] adapted T5 and LLaMa test to use kit (#4049) 1 year ago
test_shard_utils.py [test] add shard util tests 1 year ago
test_with_torch_ddp.py [shardformer] support lazy init (#4202) 1 year ago