Skip to content

Commit 041a4b2

Browse files
committed
add unittest for dataset;test=develop
1 parent 094f147 commit 041a4b2

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

python/paddle/fluid/tests/unittests/test_communicator_ps_gpu.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -74,7 +74,7 @@ def test_communicator_ps_gpu(self):
7474
batch_size=32, thread_num=1, pipe_command="cat", use_var=slots_vars)
7575
dataset.set_filelist(["test_communicator_ps_gpu.txt"])
7676
dataset._set_use_ps_gpu(1)
77-
dataset.load_into_memory()
77+
dataset.load_into_memory(is_shuffle=True)
7878

7979
os.environ["TEST_MODE"] = "1"
8080
exe = fluid.Executor(fluid.CPUPlace())

0 commit comments

Comments
 (0)