mirror of https://github.com/hpcaitech/ColossalAI
![]() * Add clip_grad_norm for hibrid_parallel_plugin * polish code * add unittests * Move tp to a higher-level optimizer interface. * bug fix * polish code |
||
---|---|---|
.. | ||
gemini | ||
low_level | ||
__init__.py | ||
wrapper.py |