ColossalAI/colossalai/interface/__init__.py

from .model import AMPModelMixin, ModelWrapper
from .optimizer import OptimizerWrapper

__all__ = ["OptimizerWrapper", "ModelWrapper", "AMPModelMixin"]
[zero] hotfix master param sync (#4618) * [zero] add method to update master params * [zero] update zero plugin * [plugin] update low level zero plugin 2023-09-05 07:04:02 +00:00			`from .model import AMPModelMixin, ModelWrapper`
[booster] implemented the torch ddd + resnet example (#3232) * [booster] implemented the torch ddd + resnet example * polish code 2023-03-27 02:24:14 +00:00			`from .optimizer import OptimizerWrapper`

[misc] update pre-commit and run all files (#4752) * [misc] update pre-commit * [misc] run pre-commit * [misc] remove useless configuration files * [misc] ignore cuda for clang-format 2023-09-19 06:20:26 +00:00			`__all__ = ["OptimizerWrapper", "ModelWrapper", "AMPModelMixin"]`