From c2377951004b459a29921a72464239ad3500f9e2 Mon Sep 17 00:00:00 2001 From: chengduo Date: Wed, 1 Aug 2018 12:14:10 +0800 Subject: [PATCH] Reopen test_parallel_executor_transformer (#12452) * reopen test_parallel_executor_transformer * save the CPU runing time --- .../tests/unittests/test_parallel_executor_transformer.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_parallel_executor_transformer.py b/python/paddle/fluid/tests/unittests/test_parallel_executor_transformer.py index bb782cf305b..8203d5d1fce 100644 --- a/python/paddle/fluid/tests/unittests/test_parallel_executor_transformer.py +++ b/python/paddle/fluid/tests/unittests/test_parallel_executor_transformer.py @@ -167,10 +167,9 @@ class TestTransformer(TestParallelExecutorBase): writer.append_tensor(t) writer.complete_append_tensor() - @unittest.skip("transformer is buggy in multi gpu") def test_main(self): 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__': -- GitLab