diff --git a/python/paddle/fluid/tests/unittests/ipu/test_weight_decay_ipu.py b/python/paddle/fluid/tests/unittests/ipu/test_weight_decay_ipu.py index 630a00f5a7d56..627a8fedff6aa 100644 --- a/python/paddle/fluid/tests/unittests/ipu/test_weight_decay_ipu.py +++ b/python/paddle/fluid/tests/unittests/ipu/test_weight_decay_ipu.py @@ -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 @@ def setUp(self): 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 @@ def exclude_fn(param): 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] diff --git a/python/paddle/fluid/tests/unittests/test_load_op_xpu.py b/python/paddle/fluid/tests/unittests/test_load_op_xpu.py index f3c8024a21ad3..f11c39b7b65df 100644 --- a/python/paddle/fluid/tests/unittests/test_load_op_xpu.py +++ b/python/paddle/fluid/tests/unittests/test_load_op_xpu.py @@ -16,6 +16,8 @@ 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 @@ def setUp(self): 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 @@ def test_load_xpu(self): 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)