提交 fa2e7dc2 编写于 作者: Q Qi Li 提交者: GitHub

Revert "fix unit test temp file, test=develop (#43155)"

This reverts commit 5a6c974e.
上级 b056c9cb
......@@ -14,9 +14,7 @@
import unittest
import os
import numpy as np
import tempfile
import paddle
import paddle.static
from paddle.fluid.tests.unittests.ipu.op_test_ipu import IPUOpTest
......@@ -32,8 +30,6 @@ class TestBase(IPUOpTest):
self.set_data_feed()
self.set_feed_attr()
self.set_attrs()
self.temp_dir = tempfile.TemporaryDirectory()
self.model_path = os.path.join(self.temp_dir.name, "weight_decay")
def set_atol(self):
self.atol = 1e-6
......@@ -92,7 +88,7 @@ class TestBase(IPUOpTest):
place = paddle.CPUPlace()
exe = paddle.static.Executor(place)
exe.run(startup_prog)
paddle.static.save(main_prog, self.model_path)
paddle.static.save(main_prog, "weight_decay")
if run_ipu:
feed_list = [image.name]
......
......@@ -16,8 +16,6 @@ from __future__ import print_function
import unittest
import numpy as np
import os
import tempfile
from op_test import OpTest, randomize_probability
import paddle.fluid as fluid
import paddle.fluid.layers as layers
......@@ -31,8 +29,6 @@ class TestLoadOpXpu(unittest.TestCase):
"""
def setUp(self):
self.temp_dir = tempfile.TemporaryDirectory()
self.model_path = os.path.join(self.temp_dir.name, "model")
self.ones = np.ones((4, 4)).astype('float32')
main_prog = fluid.Program()
start_prog = fluid.Program()
......@@ -48,7 +44,7 @@ class TestLoadOpXpu(unittest.TestCase):
exe = fluid.Executor(fluid.XPUPlace(0))
exe.run(start_prog)
fluid.io.save_persistables(exe,
dirname=self.model_path,
dirname="./model",
main_program=main_prog)
def test_load_xpu(self):
......@@ -56,7 +52,7 @@ class TestLoadOpXpu(unittest.TestCase):
start_prog = fluid.Program()
with fluid.program_guard(main_prog, start_prog):
var = layers.create_tensor(dtype='float32')
layers.load(var, file_path=self.model_path + '/w')
layers.load(var, file_path='./model/w')
exe = fluid.Executor(fluid.XPUPlace(0))
exe.run(start_prog)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册