diff --git a/PaddleNLP/similarity_net/run_classifier.py b/PaddleNLP/similarity_net/run_classifier.py index 5efd968078d383e82ddd04d0c8561c056ff76fdc..b5ec3d8acfbbd8cd55add33c2c7a41c5b259bd60 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)