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
flybird11111 eedaa3e1ef
[shardformer]fix gpt2 double head (#4663)
1 year ago
..
test_layer [shardformer] Add overlap support for gpt2 (#4535) 1 year ago
test_model [shardformer]fix gpt2 double head (#4663) 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