From f354300cda492c5a5e724d3c65eb576f732c6d2a Mon Sep 17 00:00:00 2001 From: zhang__sss Date: Fri, 30 Apr 2021 15:23:42 +0800 Subject: [PATCH] quant_mode 2 --- mindspore/train/serialization.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/mindspore/train/serialization.py b/mindspore/train/serialization.py index 191686d795..f9d3583573 100644 --- a/mindspore/train/serialization.py +++ b/mindspore/train/serialization.py @@ -756,9 +756,11 @@ def quant_mode_manage(func): Inherit the quant_mode in old version. """ def warpper(network, *inputs, file_format, **kwargs): - if not kwargs.get('quant_mode', None): + if 'quant_mode' not in kwargs: return network quant_mode = kwargs['quant_mode'] + if not isinstance(quant_mode, str): + raise TypeError("The type of quant_mode should be str, but got {}.".format(type(quant_mode))) if quant_mode in ('AUTO', 'MANUAL'): kwargs['quant_mode'] = 'QUANT' return func(network, *inputs, file_format=file_format, **kwargs)