ColossalAI/colossalai/nn/parallel/layers/cache_embedding
Jiarui Fang 521078ffc9
[embedding] fix a bug in table wise sharding (#1538)
2022-09-02 15:48:35 +08:00
..
__init__.py
base_embedding.py
cache_mgr.py
copyer.py
freq_aware_embedding.py
parallel_freq_aware_embedding.py
parallel_freq_aware_embedding_tablewise.py [embedding] fix a bug in table wise sharding (#1538) 2022-09-02 15:48:35 +08:00