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

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

上级 7e439780
......@@ -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(
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册