mindspore_lite.ModelType

class mindspore_lite.ModelType[源代码]

ModelType 类定义MindSpore Lite中导出或导入的模型类型。

适用于以下场景：

  1. Converter时，设置 export_mindir 参数， ModelType 用于定义转换生成的模型类型。

  2. Converter之后，当从文件加载或构建模型以进行推理时， ModelType 用于定义输入模型框架类型。

目前，支持以下 ModelType ：

定义

说明

ModelType.MINDIR

MindSpore模型的框架类型，该模型使用.mindir作为后缀。

ModelType.MINDIR_LITE

MindSpore Lite模型的框架类型，该模型使用.ms作为后缀。

样例：

>>> # Method 1: Import mindspore_lite package
>>> import mindspore_lite as mslite
>>> print(mslite.ModelType.MINDIR_LITE)
ModelType.MINDIR_LITE
>>> # Method 2: from mindspore_lite package import ModelType
>>> from mindspore_lite import ModelType
>>> print(ModelType.MINDIR_LITE)
ModelType.MINDIR_LITE