github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/nn/layer/__init__.py

3 lines
40 B
Python
Raw Normal View History

[MOE] changed parallelmode to dist process group (#460)
2022-03-19 05:46:29 +00:00
from .moe import *
Hotfix/Colossalai layers (#92) * optimized 1d layer apis; reorganized nn.layer modules; fixed tests * fixed 2.5d runtime issue * reworked split batch, now called in trainer.schedule.load_batch Co-authored-by: BoxiangW <45734921+BoxiangW@users.noreply.github.com>
2021-12-29 15:32:10 +00:00
from .utils import *