From e64168146dd2c2c0d33eb5e2784759c96e4b7861 Mon Sep 17 00:00:00 2001 From: zhangyi Date: Tue, 9 Mar 2021 11:25:15 +0800 Subject: [PATCH] fix error format for api comments. --- mindspore/train/model.py | 1 + mindspore/train/serialization.py | 2 ++ 2 files changed, 3 insertions(+) diff --git a/mindspore/train/model.py b/mindspore/train/model.py index 0edaaa77a8..8d4cab2e52 100755 --- a/mindspore/train/model.py +++ b/mindspore/train/model.py @@ -76,6 +76,7 @@ class Model: to other metric. Default: None. amp_level (str): Option for argument `level` in `mindspore.amp.build_train_network`, level for mixed precision training. Supports ["O0", "O2", "O3", "auto"]. Default: "O0". + - O0: Do not change. - O2: Cast network to float16, keep batchnorm run in float32, using dynamic loss scale. - O3: Cast network to float16, with additional property 'keep_batchnorm_fp32=False'. diff --git a/mindspore/train/serialization.py b/mindspore/train/serialization.py index 283ad875b2..f018c0cfda 100644 --- a/mindspore/train/serialization.py +++ b/mindspore/train/serialization.py @@ -580,6 +580,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 Intermediate Representation. An intermediate 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. @@ -589,6 +590,7 @@ def export(net, *inputs, file_name, file_format='AIR', **kwargs): 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.