mirror of https://github.com/hpcaitech/ColossalAI
707b11d4a0
* [zero] add strict ddp mode for chunk init * [gemini] update gpt example |
||
---|---|---|
.. | ||
layers | ||
__init__.py | ||
data_parallel.py | ||
gemini_parallel.py | ||
reducer.py | ||
utils.py |