提交 19a8d965 编写于 作者: J JiabinYang

fix nce in test_dist_transpiler, test=develop

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