From 19a8d965858173789376248b076fc0339422d313 Mon Sep 17 00:00:00 2001 From: JiabinYang Date: Tue, 18 Dec 2018 13:18:11 +0000 Subject: [PATCH] fix nce in test_dist_transpiler, test=develop --- python/paddle/fluid/tests/unittests/test_dist_transpiler.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_dist_transpiler.py b/python/paddle/fluid/tests/unittests/test_dist_transpiler.py index 73795a2154..0555db4cba 100644 --- a/python/paddle/fluid/tests/unittests/test_dist_transpiler.py +++ b/python/paddle/fluid/tests/unittests/test_dist_transpiler.py @@ -871,8 +871,8 @@ class TestRemoteNce(TestDistLookupTableBase): def transpiler_test_impl(self): trainer, _ = self.get_trainer() - out_vars = ["nce_w.block0", "nce_w.block1"] - in_vars = ["nce_b.block0", "nce_b.block1"] + out_vars = ["nce_w"] + in_vars = ["nce_b"] recv_var_names = []