Skip to content

P
Paddle

机器未来 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 b70002c5 编写于 作者: S seiriosPlus
浏览文件
操作

fix UT

上级 ef2b1af4
显示空白变更内容
内联 并排
Showing with 2 addition and 2 deletion
python/paddle/fluid/tests/unittests/test_dist_fleet_ps4.py
浏览文件 @ b70002c5
...@@ -68,7 +68,7 @@ class TestPSPassWithBow(unittest.TestCase): ...@@ -68,7 +68,7 @@ class TestPSPassWithBow(unittest.TestCase):
name="query_ids", shape=[1], dtype="int64", lod_level=1) name="query_ids", shape=[1], dtype="int64", lod_level=1)
# embedding # embedding
q_emb = fluid.contrib.layers.sparse_embedding( q_emb = fluid.contrib.layers.sparse_embedding(
input=pt, input=q,
size=[dict_dim, emb_dim], size=[dict_dim, emb_dim],
param_attr=fluid.ParamAttr( param_attr=fluid.ParamAttr(
initializer=fluid.initializer.Constant(value=0.01), initializer=fluid.initializer.Constant(value=0.01),
...@@ -117,7 +117,7 @@ class TestPSPassWithBow(unittest.TestCase): ...@@ -117,7 +117,7 @@ class TestPSPassWithBow(unittest.TestCase):
name="neg_title_ids", shape=[1], dtype="int64", lod_level=1) name="neg_title_ids", shape=[1], dtype="int64", lod_level=1)
# embedding # embedding
nt_emb = fluid.contrib.layers.sparse_embedding( nt_emb = fluid.contrib.layers.sparse_embedding(
input=pt, input=nt,
size=[dict_dim, emb_dim], size=[dict_dim, emb_dim],
param_attr=fluid.ParamAttr( param_attr=fluid.ParamAttr(
initializer=fluid.initializer.Constant(value=0.01), initializer=fluid.initializer.Constant(value=0.01),
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或