mirror of https://github.com/hpcaitech/ColossalAI
6ef33f75aa
* support DDP for HybridPlugin/add tp+dp tests * add docstring for HybridParallelPlugin |
||
---|---|---|
.. | ||
mixed_precision | ||
plugin | ||
__init__.py | ||
accelerator.py | ||
booster.py |