mirror of https://github.com/hpcaitech/ColossalAI
83b52c56cd
* Add clip_grad_norm for hibrid_parallel_plugin * polish code * add unittests * Move tp to a higher-level optimizer interface. * bug fix * polish code |
||
---|---|---|
.. | ||
naive_amp | ||
__init__.py |