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 7486ed7d3a
[shardformer] update llama2/opt finetune example and fix llama2 policy (#4645)
1 year ago
..
test_layer [shardformer] Add overlap support for gpt2 (#4535) 1 year ago
test_model [shardformer] update llama2/opt finetune example and fix llama2 policy (#4645) 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