From ab1d8b838708c81643d24280eb708e961bc15672 Mon Sep 17 00:00:00 2001 From: Lizhengo <18811528909@163.com> Date: Mon, 5 Aug 2019 20:45:55 +0800 Subject: [PATCH] fix pred in simnet (#2995) --- PaddleNLP/similarity_net/run_classifier.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/PaddleNLP/similarity_net/run_classifier.py b/PaddleNLP/similarity_net/run_classifier.py index 5efd9680..b5ec3d8a 100644 --- a/PaddleNLP/similarity_net/run_classifier.py +++ b/PaddleNLP/similarity_net/run_classifier.py @@ -138,7 +138,7 @@ def train(conf_dict, args): valid_feeder = fluid.DataFeeder( place=place, feed_list=[left.name, pos_right.name]) valid_reader = simnet_process.get_reader("valid") - pred = pos_score + pred = pos_score # Save Infer model infer_program = fluid.default_main_program().clone(for_test=True) _, neg_score = net.predict(left, neg_right) -- GitLab