fix nce in test_dist_transpiler, test=develop

6 years ago · 19a8d96585
parent 52ca40fcfe
commit 19a8d96585
1 changed files with 2 additions and 2 deletions
--- 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 = []