mindspore_lite.ModelType

class mindspore_lite.ModelType[source]

The ModelType class defines the type of the model exported or imported in MindSpot Lite.

Used in the following scenarios:

1. When using mindspore_lite.Converter, set save_type parameter, ModelType used to define the model type generated by Converter. ModelType.MINDIR is recommended.

2. After using mindspore_lite.Converter, when loading or building a model from file for predicting, the ModelType is used to define Input model framework type. Only support ModelType.MINDIR.

Currently, the following ModelType are supported:

Definition	Description
ModelType.MINDIR	MindSpore model’s framework type, which model uses .mindir as suffix.
ModelType.MINDIR_LITE	MindSpore Lite model’s framework type, which model uses .ms as suffix.

Examples

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