未验证 提交 56cd1eca 编写于 作者: L Liufang Sang 提交者: GitHub

add power to quant_embedding dict (#289)

上级 9a722e53
......@@ -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(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册