diff --git a/paddleslim/quant/quant_embedding.py b/paddleslim/quant/quant_embedding.py index c5c3bc62a7aaddafce11151fdaa0d49ab1f8c05a..2e4f229859097ec26d849f0f2c8839ba80b8c164 100755 --- a/paddleslim/quant/quant_embedding.py +++ b/paddleslim/quant/quant_embedding.py @@ -334,6 +334,7 @@ def _quant_embedding_log(graph, scope, place, config, var_name, # get quantize dict and quanted tensor topk_num, quanted_tensor = _quant_log(embedding_tensor, config) + topk_num = np.power(2, topk_num) #create params must use create_persistable_node topk_num_var = graph.create_persistable_node(