未验证 提交 5a6c974e 编写于 作者: Q Qi Li 提交者: GitHub

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

上级 9c17688a
......@@ -14,7 +14,9 @@
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
......@@ -30,6 +32,8 @@ 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
......@@ -88,7 +92,7 @@ class TestBase(IPUOpTest):
place = paddle.CPUPlace()
exe = paddle.static.Executor(place)
exe.run(startup_prog)
paddle.static.save(main_prog, "weight_decay")
paddle.static.save(main_prog, self.model_path)
if run_ipu:
feed_list = [image.name]
......
......@@ -16,6 +16,8 @@ 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
......@@ -29,6 +31,8 @@ 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()
......@@ -44,7 +48,7 @@ class TestLoadOpXpu(unittest.TestCase):
exe = fluid.Executor(fluid.XPUPlace(0))
exe.run(start_prog)
fluid.io.save_persistables(exe,
dirname="./model",
dirname=self.model_path,
main_program=main_prog)
def test_load_xpu(self):
......@@ -52,7 +56,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='./model/w')
layers.load(var, file_path=self.model_path + '/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.
先完成此消息的编辑!
想要评论请 注册