ColossalAI/tests/test_shardformer/test_layer
FoolPlayer 507c0ad368 add vocabembedding layer 2023-07-04 16:05:01 +08:00
..
test_dropout.py [shardformer] refactored embedding and dropout to parallel module (#4013) 2023-07-04 16:05:01 +08:00
test_embedding.py [shardformer] refactored embedding and dropout to parallel module (#4013) 2023-07-04 16:05:01 +08:00
test_linear_1d.py [shardformer] refactored embedding and dropout to parallel module (#4013) 2023-07-04 16:05:01 +08:00
test_vocab_parallel_embedding_1d.py add vocabembedding layer 2023-07-04 16:05:01 +08:00