Merge pull request #212 from Superjom/fix_dssm

Fix dssm infer error with gru

dssm/infer.py

@@ -94,7 +94,7 @@ class Inferer(object):
     def __init__(self, param_path):
         logger.info("create DSSM model")
 
-        cost, prediction, label = DSSM(
+        prediction = DSSM(
             dnn_dims=layer_dims,
             vocab_sizes=[
                 len(load_dic(path))
@@ -104,7 +104,8 @@ class Inferer(object):
             model_arch=args.model_arch,
             share_semantic_generator=args.share_network_between_source_target,
             class_num=args.class_num,
-            share_embed=args.share_embed)()
+            share_embed=args.share_embed,
+            is_infer=True)()
 
         # load parameter
         logger.info("load model parameters from %s" % param_path)

dssm/network_conf.py

@@ -102,8 +102,7 @@ class DSSM(object):
         '''
         A GRU sentence vector learner.
         '''
-        gru = paddle.layer.gru_memory(
-            input=emb, )
+        gru = paddle.networks.simple_gru(input=emb, size=256)
         sent_vec = paddle.layer.last_seq(gru)
         return sent_vec
 
@@ -219,7 +218,7 @@ class DSSM(object):
             # but this operator is not supported currently.
             # so AUC will not used.
             return cost, None, label
-        return None, [left_score, right_score], label
+        return right_score
 
     def _build_classification_or_regression_model(self, is_classification):
         '''
@@ -275,4 +274,4 @@ class DSSM(object):
 
         if not self.is_infer:
             return cost, prediction, label
-        return None, prediction, label
+        return prediction