diff --git a/mindspore/train/serialization.py b/mindspore/train/serialization.py index c4d35b4c80..5220f4a8fd 100644 --- a/mindspore/train/serialization.py +++ b/mindspore/train/serialization.py @@ -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.