Browse Source

!7547 mode_export_note

Merge pull request !7547 from baiyangfan/mode_export_note
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
eaab26f7a9
1 changed files with 3 additions and 0 deletions
  1. +3
    -0
      mindspore/train/serialization.py

+ 3
- 0
mindspore/train/serialization.py View File

@@ -501,6 +501,7 @@ def export(net, *inputs, file_name, file_format='AIR', **kwargs):
inputs (Tensor): Inputs of the `net`.
file_name (str): File name of the model to be exported.
file_format (str): MindSpore currently supports 'AIR', 'ONNX' and 'MINDIR' format for exported model.

- AIR: Ascend Intermidiate Representation. An intermidiate representation format of Ascend model.
Recommended suffix for output file is '.air'.
- ONNX: Open Neural Network eXchange. An open format built to represent machine learning models.
@@ -508,7 +509,9 @@ def export(net, *inputs, file_name, file_format='AIR', **kwargs):
- MINDIR: MindSpore Native Intermidiate Representation for Anf. An intermidiate representation format
for MindSpore models.
Recommended suffix for output file is '.mindir'.

kwargs (dict): Configuration options dictionary.

- quant_mode: The mode of quant.
- mean: Input data mean. Default: 127.5.
- std_dev: Input data variance. Default: 127.5.


Loading…
Cancel
Save