mirror of https://github.com/hpcaitech/ColossalAI
51b9a49655
* add zero optimizer * torch ok * unit test ok * polish code * fix bugs * polish unit test * polish zero optim * polish colo ddp v2 * refactor folder structure * add comment * polish unit test * polish zero optim * polish unit test |
||
---|---|---|
.. | ||
__init__.py | ||
zero_hook.py | ||
zero_hook_v2.py |