From 041a4b2fbefc4a8cff735ac68dfb41a1141bfc75 Mon Sep 17 00:00:00 2001 From: danleifeng Date: Mon, 5 Jul 2021 08:02:48 +0000 Subject: [PATCH] add unittest for dataset;test=develop --- python/paddle/fluid/tests/unittests/test_communicator_ps_gpu.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/fluid/tests/unittests/test_communicator_ps_gpu.py b/python/paddle/fluid/tests/unittests/test_communicator_ps_gpu.py index 0b956d5031fec..6ab8a2c3a4b22 100644 --- a/python/paddle/fluid/tests/unittests/test_communicator_ps_gpu.py +++ b/python/paddle/fluid/tests/unittests/test_communicator_ps_gpu.py @@ -74,7 +74,7 @@ def test_communicator_ps_gpu(self): batch_size=32, thread_num=1, pipe_command="cat", use_var=slots_vars) dataset.set_filelist(["test_communicator_ps_gpu.txt"]) dataset._set_use_ps_gpu(1) - dataset.load_into_memory() + dataset.load_into_memory(is_shuffle=True) os.environ["TEST_MODE"] = "1" exe = fluid.Executor(fluid.CPUPlace())