Browse Source

fix_export_bug

tags/v1.1.0
bai-yangfan 5 years ago
parent
commit
8531587cfe
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