ColossalAI/tests/test_shardformer
FoolPlayer 74d176c8d8 [shardformer] fix bert and gpt downstream with new api (#4024)
* fix bert downstream with new api

* remove comment line
2023-07-04 16:05:01 +08:00
..
test_layer add vocabembedding layer 2023-07-04 16:05:01 +08:00
test_model [shardformer] fix bert and gpt downstream with new api (#4024) 2023-07-04 16:05:01 +08:00
test_module [shardformer] Unit test (#3928) 2023-07-04 16:05:01 +08:00