未验证 提交 f2178376 编写于 作者: Z Zeng Jinle 提交者: GitHub

fix gc transformer ut, test=develop (#23421)

上级 7e439780
...@@ -25,6 +25,8 @@ import paddle.dataset.wmt16 as wmt16 ...@@ -25,6 +25,8 @@ import paddle.dataset.wmt16 as wmt16
import os import os
from feed_data_reader import FeedDataReader from feed_data_reader import FeedDataReader
os.environ['CPU_NUM'] = str(4)
class ModelHyperParams(object): class ModelHyperParams(object):
# Dictionary size for source and target language. This model directly uses # Dictionary size for source and target language. This model directly uses
...@@ -185,10 +187,6 @@ def get_feed_data_reader(): ...@@ -185,10 +187,6 @@ def get_feed_data_reader():
class TestTransformer(TestParallelExecutorBase): class TestTransformer(TestParallelExecutorBase):
@classmethod
def setUpClass(cls):
os.environ['CPU_NUM'] = str(4)
def test_main(self): def test_main(self):
if core.is_compiled_with_cuda(): if core.is_compiled_with_cuda():
self.check_network_convergence( self.check_network_convergence(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册