ColossalAI/colossalai/nn/parallel/layers/cache_embedding
Jiarui Fang a19eb80998
[embedding] updates some default parameters
2022-09-15 15:45:17 +08:00
..
__init__.py [embedding] polish parallel embedding tablewise (#1545) 2022-09-06 10:41:20 +08:00
base_embedding.py
cache_mgr.py [embedding] updates some default parameters 2022-09-15 15:45:17 +08:00
copyer.py [NFC] polish doc style for ColoTensor (#1457) 2022-08-16 09:21:05 +08:00
embedding_config.py [embedding] polish parallel embedding tablewise (#1545) 2022-09-06 10:41:20 +08:00
freq_aware_embedding.py [embedding] updates some default parameters 2022-09-15 15:45:17 +08:00
parallel_freq_aware_embedding.py [embedding] cache_embedding small improvement (#1564) 2022-09-08 16:41:19 +08:00
parallel_freq_aware_embedding_tablewise.py [embedding] updates some default parameters 2022-09-15 15:45:17 +08:00
parallel_freq_aware_embedding_tablewise_split_cache.py [embedding] polish parallel embedding tablewise (#1545) 2022-09-06 10:41:20 +08:00