diff --git a/paddleslim/quant/quant_embedding.py b/paddleslim/quant/quant_embedding.py index 7090ea0a307c5c100ad8b1e9d85aa05baf7e5eaf..c49b6762b1844026e0d6dfaaab2a36108fe2c7cd 100755 --- a/paddleslim/quant/quant_embedding.py +++ b/paddleslim/quant/quant_embedding.py @@ -70,7 +70,7 @@ def _merge_config(old_config, new_config): supported, now supported dtypes are {} \ ".format(dtype, support_dtype) if 'threshold' in keys: - assert isinstance(new_config['threshold'], (float, int)), "threshold \ + assert isinstance(old_config['threshold'], (float, int)), "threshold \ must be number." print("quant_embedding config {}".format(old_config))