From f217837654d5eff4f158679e7bd8dee4563da17c Mon Sep 17 00:00:00 2001
From: Zeng Jinle <32832641+sneaxiy@users.noreply.github.com>
Date: Thu, 2 Apr 2020 07:48:38 -0500
Subject: [PATCH] fix gc transformer ut, test=develop (#23421)

---
 .../tests/unittests/test_parallel_executor_transformer.py   | 6 ++----
 1 file changed, 2 insertions(+), 4 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 1f47d87811c..2d1e0e98498 100644
--- a/python/paddle/fluid/tests/unittests/test_parallel_executor_transformer.py
+++ b/python/paddle/fluid/tests/unittests/test_parallel_executor_transformer.py
@@ -25,6 +25,8 @@ import paddle.dataset.wmt16 as wmt16
 import os
 from feed_data_reader import FeedDataReader
 
+os.environ['CPU_NUM'] = str(4)
+
 
 class ModelHyperParams(object):
     # Dictionary size for source and target language. This model directly uses
@@ -185,10 +187,6 @@ def get_feed_data_reader():
 
 
 class TestTransformer(TestParallelExecutorBase):
-    @classmethod
-    def setUpClass(cls):
-        os.environ['CPU_NUM'] = str(4)
-
     def test_main(self):
         if core.is_compiled_with_cuda():
             self.check_network_convergence(
-- 
GitLab