mirror of https://github.com/hpcaitech/ColossalAI
108e54a0b4
* [shardformer] gpt2 tests fix [shardformer] test all optimizations (#4399) [shardformer] test all optimizations [shardformer] test all optimizations [shardformer] test all optimizations [shardformer] gpt2 tests fix * [shardformer]update t5 to use all optimizations |
||
---|---|---|
.. | ||
test_layer | ||
test_model | ||
__init__.py | ||
test_shard_utils.py | ||
test_with_torch_ddp.py |