mirror of https://github.com/hpcaitech/ColossalAI
![]() * [booster] add prepare dataloader method for plug * [booster] update examples and docstr |
||
---|---|---|
.. | ||
__init__.py | ||
dp_plugin_base.py | ||
gemini_plugin.py | ||
low_level_zero_plugin.py | ||
plugin_base.py | ||
torch_ddp_plugin.py |