OSSInnovation
/
mindspore
1
0
Fork 26
Code Issues Pull Requests 7 Releases Wiki Activity

!6715 modify quant export annotation 

Merge pull request !6715 from yuchaojie/r1
This commit is contained in:
mindspore-ci-bot 2020-09-22 18:38:35 +08:00 committed by Gitee
parent 0bb1bfab81 b270671c39
commit 8d2a936eee
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mindspore/train/quant/quant.py
View File

@ -487,9 +487,9 @@ def export(network, *inputs, file_name, mean=127.5, std_dev=127.5, file_format='
network (Cell): MindSpore network produced by `convert_quant_network`.
inputs (Tensor): Inputs of the `quantization aware training network`.
file_name (str): File name of model to export.
mean (int): Input data mean. Default: 127.5.
mean (int, float): Input data mean. Default: 127.5.
std_dev (int, float): Input data variance. Default: 127.5.
file_format (str): MindSpore currently supports 'AIR', 'ONNX' and 'MINDIR' format for exported
file_format (str): MindSpore currently supports 'AIR' and 'MINDIR' format for exported
quantization aware model. Default: 'AIR'.
- AIR: Graph Engine Intermidiate Representation. An intermidiate representation format of