ColossalAI/colossalai/nn/parallel
Jiarui Fang a19eb80998
[embedding] updates some default parameters
2022-09-15 15:45:17 +08:00
..
layers [embedding] updates some default parameters 2022-09-15 15:45:17 +08:00
__init__.py
data_parallel.py [NFC] polish colossalai/nn/parallel/data_parallel.py code style (#1570) 2022-09-08 22:11:04 +08:00
reducer.py [ddp] ColoDDP uses bucket all-reduce (#1177) 2022-06-29 10:34:13 +08:00