Files
ColossalAI/colossalai/booster/plugin/plugin_base.py
Frank Lee e7f3bed2d3 [booster] added the plugin base and torch ddp plugin (#3180)
* [booster] added the plugin base and torch ddp plugin

* polish code

* polish code

* polish code
2023-03-21 17:39:30 +08:00

1.1 KiB