Make test_parallel_executor_seresnet.py Faster (#16701)

* slimming test_parallel_executor_seresnet.py
7 years ago · 610c6442e3
parent 112f16143b
commit 610c6442e3
2 changed files with 120 additions and 132 deletions
--- a/python/paddle/fluid/tests/unittests/parallel_executor_test_base.py
+++ b/python/paddle/fluid/tests/unittests/parallel_executor_test_base.py
@ -29,7 +29,8 @@ __all__ = ['TestParallelExecutorBase']


 class TestParallelExecutorBase(unittest.TestCase):
-    def check_network_convergence(self,
+    @classmethod
+    def check_network_convergence(cls,
                                  method,
                                  use_cuda=True,
                                  memory_opt=True,
--- a/python/paddle/fluid/tests/unittests/test_parallel_executor_seresnext.py
+++ b/python/paddle/fluid/tests/unittests/test_parallel_executor_seresnext.py