quant_mode 2

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)