未验证 提交 c2377951 编写于 作者: C chengduo 提交者: GitHub

Reopen test_parallel_executor_transformer (#12452)

* reopen test_parallel_executor_transformer

* save the CPU runing time
上级 b5a3e40b
...@@ -167,10 +167,9 @@ class TestTransformer(TestParallelExecutorBase): ...@@ -167,10 +167,9 @@ class TestTransformer(TestParallelExecutorBase):
writer.append_tensor(t) writer.append_tensor(t)
writer.complete_append_tensor() writer.complete_append_tensor()
@unittest.skip("transformer is buggy in multi gpu")
def test_main(self): def test_main(self):
self.check_network_convergence(transformer, use_cuda=True) self.check_network_convergence(transformer, use_cuda=True)
self.check_network_convergence(transformer, use_cuda=False) self.check_network_convergence(transformer, use_cuda=False, iter=5)
if __name__ == '__main__': if __name__ == '__main__':
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册