Skip to content

Commit

Permalink
modify test_load_op save path from /tmp to ./ (#27872)
Browse files Browse the repository at this point in the history
* modify test_load_op save path from /tmp to ./, test=develop

* trigger kunlun ci, test=kunlun

* trigger kunlun ci, test=kunlun
  • Loading branch information
wanghuancoder authored Oct 13, 2020
1 parent 9637d96 commit 92708a2
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
4 changes: 2 additions & 2 deletions python/paddle/fluid/tests/unittests/test_load_op.py
Original file line number Diff line number Diff line change
Expand Up @@ -41,14 +41,14 @@ def setUp(self):
exe = fluid.Executor(fluid.CPUPlace())
exe.run(start_prog)
fluid.io.save_persistables(
exe, dirname="/tmp/model", main_program=main_prog)
exe, dirname="./model", main_program=main_prog)

def test_load(self):
main_prog = fluid.Program()
start_prog = fluid.Program()
with fluid.program_guard(main_prog, start_prog):
var = layers.create_tensor(dtype='float32')
layers.load(var, file_path='/tmp/model/w')
layers.load(var, file_path='./model/w')

exe = fluid.Executor(fluid.CPUPlace())
exe.run(start_prog)
Expand Down
4 changes: 2 additions & 2 deletions python/paddle/fluid/tests/unittests/test_load_op_xpu.py
Original file line number Diff line number Diff line change
Expand Up @@ -44,14 +44,14 @@ def setUp(self):
exe = fluid.Executor(fluid.XPUPlace(0))
exe.run(start_prog)
fluid.io.save_persistables(
exe, dirname="/tmp/model", main_program=main_prog)
exe, dirname="./model", main_program=main_prog)

def test_load_xpu(self):
main_prog = fluid.Program()
start_prog = fluid.Program()
with fluid.program_guard(main_prog, start_prog):
var = layers.create_tensor(dtype='float32')
layers.load(var, file_path='/tmp/model/w')
layers.load(var, file_path='./model/w')

exe = fluid.Executor(fluid.XPUPlace(0))
exe.run(start_prog)
Expand Down

0 comments on commit 92708a2

Please sign in to comment.