mirror of
https://github.com/hpcaitech/ColossalAI.git
synced 2025-08-23 10:11:37 +00:00
807e01a4ba
* [zero] add method to update master params * [zero] update zero plugin * [plugin] update low level zero plugin
5 lines
152 B
Python
5 lines
152 B
Python
from .model import AMPModelMixin, ModelWrapper
|
|
from .optimizer import OptimizerWrapper
|
|
|
|
__all__ = ['OptimizerWrapper', 'ModelWrapper', 'AMPModelMixin']
|