Browse Source

!7636 fix_export_bug

Merge pull request !7636 from baiyangfan/mode_export_bug
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
bc1d33c62d
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindspore/train/serialization.py

+ 2
- 2
mindspore/train/serialization.py View File

@@ -570,8 +570,8 @@ def _quant_export(network, *inputs, file_format, **kwargs):
supported_formats = ['AIR', 'MINDIR']
quant_mode_formats = ['AUTO', 'MANUAL']

mean = kwargs['mean'] if kwargs.get('mean', None) else 127.5
std_dev = kwargs['std_dev'] if kwargs.get('std_dev', None) else 127.5
mean = 127.5 if kwargs.get('mean', None) is None else kwargs['mean']
std_dev = 127.5 if kwargs.get('std_dev', None) is None else kwargs['std_dev']

quant_mode = kwargs['quant_mode']
if quant_mode not in quant_mode_formats:


Loading…
Cancel
Save