mirror of https://github.com/hpcaitech/ColossalAI
ea13a201bb
* [gemini] polish code * [testing] remove code * [gemini] make more robust |
||
---|---|---|
.. | ||
layers | ||
__init__.py | ||
data_parallel.py | ||
gemini_parallel.py | ||
reducer.py | ||
utils.py |