From c46252acaf6382ca249d421c92d4ae340cd6dc4c Mon Sep 17 00:00:00 2001 From: malin10 Date: Tue, 18 Aug 2020 12:08:01 +0800 Subject: [PATCH] bug fix --- models/recall/word2vec/model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/models/recall/word2vec/model.py b/models/recall/word2vec/model.py index 4b19513d..8775e08b 100755 --- a/models/recall/word2vec/model.py +++ b/models/recall/word2vec/model.py @@ -209,7 +209,7 @@ class Model(ModelBase): emb_all_label_l2 = fluid.layers.l2_normalize(x=emb_all_label, axis=1) dist = fluid.layers.matmul( x=target, y=emb_all_label_l2, transpose_y=True) - values, pred_idx = fluid.layers.topk(input=dist, 1) + values, pred_idx = fluid.layers.topk(input=dist, k=1) label = fluid.layers.expand( fluid.layers.unsqueeze( inputs[3], axes=[1]), expand_times=[1, 1]) -- GitLab