From d0fd18cff3f4e2d6fe61edf3a1a5f72fc02cead2 Mon Sep 17 00:00:00 2001 From: yuchaojie Date: Tue, 22 Sep 2020 15:07:16 +0800 Subject: [PATCH] modify quant export annotation --- mindspore/train/quant/quant.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/mindspore/train/quant/quant.py b/mindspore/train/quant/quant.py index 9d64704d81..74aaf132a6 100644 --- a/mindspore/train/quant/quant.py +++ b/mindspore/train/quant/quant.py @@ -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