提交 1506c808 编写于 作者: S Superjom

fix dssm gru

上级 47fb064e
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册