mirror of https://github.com/hpcaitech/ColossalAI
![]() * [shardformer] refactored embedding and dropout to parallel module * polish code |
||
---|---|---|
.. | ||
test_dropout.py | ||
test_embedding.py | ||
test_linear_1d.py |
![]() * [shardformer] refactored embedding and dropout to parallel module * polish code |
||
---|---|---|
.. | ||
test_dropout.py | ||
test_embedding.py | ||
test_linear_1d.py |