diff --git a/dssm/network_conf.py b/dssm/network_conf.py index f0f2e1a209debe6bfd3c826048c0afc0b5f9dd66..430db232db7734368d1261adcaef87a1360c0f28 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