:py:mod:`neural_compressor.model` ================================= .. py:module:: neural_compressor.model .. autoapi-nested-parse:: Built-in model for multiple framework backends. Submodules ---------- .. toctree:: :titlesonly: :maxdepth: 1 base_model/index.rst keras_model/index.rst model/index.rst mxnet_model/index.rst nets_factory/index.rst onnx_model/index.rst tensorflow_model/index.rst torch_model/index.rst Package Contents ---------------- Classes ~~~~~~~ .. autoapisummary:: neural_compressor.model.Model neural_compressor.model.BaseModel Attributes ~~~~~~~~~~ .. autoapisummary:: neural_compressor.model.MODELS .. py:class:: Model Bases: :py:obj:`object` A wrapper to construct a Neural Compressor Model. .. py:class:: BaseModel(model, **kwargs) Base class of all neural_compressor.model, will play graph role. .. py:property:: model :abstractmethod: Return model itself. .. py:property:: graph_info :abstractmethod: Node_type'. :type: Return a dict with content 'Node .. py:method:: save(root, *args, **kwargs) :abstractmethod: Abstract method of model saving. .. py:method:: export(save_path: str, conf) :abstractmethod: Abstract method of model convertion to ONNX. .. py:method:: framework() :abstractmethod: Abstract method of model framework.