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