From b70002c59f475457132bd51bdca7765803b4d3e0 Mon Sep 17 00:00:00 2001 From: seiriosPlus Date: Mon, 21 Sep 2020 18:53:27 +0800 Subject: [PATCH] fix UT --- python/paddle/fluid/tests/unittests/test_dist_fleet_ps4.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_dist_fleet_ps4.py b/python/paddle/fluid/tests/unittests/test_dist_fleet_ps4.py index 88ed835d796..f1176aea34e 100644 --- a/python/paddle/fluid/tests/unittests/test_dist_fleet_ps4.py +++ b/python/paddle/fluid/tests/unittests/test_dist_fleet_ps4.py @@ -68,7 +68,7 @@ class TestPSPassWithBow(unittest.TestCase): name="query_ids", shape=[1], dtype="int64", lod_level=1) # embedding q_emb = fluid.contrib.layers.sparse_embedding( - input=pt, + input=q, size=[dict_dim, emb_dim], param_attr=fluid.ParamAttr( initializer=fluid.initializer.Constant(value=0.01), @@ -117,7 +117,7 @@ class TestPSPassWithBow(unittest.TestCase): name="neg_title_ids", shape=[1], dtype="int64", lod_level=1) # embedding nt_emb = fluid.contrib.layers.sparse_embedding( - input=pt, + input=nt, size=[dict_dim, emb_dim], param_attr=fluid.ParamAttr( initializer=fluid.initializer.Constant(value=0.01), -- GitLab