ColossalAI/tests/test_shardformer
FoolPlayer 4021b9a8a2 [shardformer] add gpt2 test and layer class refactor (#4041)
* add gpt2 test and layer class refactor

* add dropout in gpt2 policy
2023-07-04 16:05:01 +08:00
..
test_layer add vocabembedding layer 2023-07-04 16:05:01 +08:00
test_model [shardformer] add gpt2 test and layer class refactor (#4041) 2023-07-04 16:05:01 +08:00
test_module [shardformer] Unit test (#3928) 2023-07-04 16:05:01 +08:00