From 1506c80854c79b3ed4dfe43ac6f0eb5f4bdc9fe8 Mon Sep 17 00:00:00 2001 From: Superjom Date: Sun, 20 Aug 2017 20:59:30 -0700 Subject: [PATCH] fix dssm gru --- dssm/network_conf.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/dssm/network_conf.py b/dssm/network_conf.py index f0f2e1a2..430db232 100644 --- a/dssm/network_conf.py +++ b/dssm/network_conf.py @@ -102,8 +102,8 @@ 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 -- GitLab