p60975123
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!7636 fix_export_bug 

Merge pull request !7636 from baiyangfan/mode_export_bug
This commit is contained in:
mindspore-ci-bot 2020-10-22 19:27:40 +08:00 committed by Gitee
parent d54ea04df3 8531587cfe
commit bc1d33c62d
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/serialization.py
View File

@ -570,8 +570,8 @@ def _quant_export(network, *inputs, file_format, **kwargs):
supported_formats = ['AIR', 'MINDIR'] supported_formats = ['AIR', 'MINDIR']
quant_mode_formats = ['AUTO', 'MANUAL'] quant_mode_formats = ['AUTO', 'MANUAL']
mean = kwargs['mean'] if kwargs.get('mean', None) else 127.5 mean = 127.5 if kwargs.get('mean', None) is None else kwargs['mean']
std_dev = kwargs['std_dev'] if kwargs.get('std_dev', None) else 127.5 std_dev = 127.5 if kwargs.get('std_dev', None) is None else kwargs['std_dev']
quant_mode = kwargs['quant_mode'] quant_mode = kwargs['quant_mode']
if quant_mode not in quant_mode_formats: if quant_mode not in quant_mode_formats: