From 96491e43b7dad85ff9a52e4336dd06ec8fc1b522 Mon Sep 17 00:00:00 2001 From: guofei <52460041+gfwm2013@users.noreply.github.com> Date: Thu, 23 Apr 2020 15:48:58 +0800 Subject: [PATCH] Repair unittest: test_dyn_rnn (#24031) Repair unittest: test_dyn_rnn --- python/paddle/fluid/tests/unittests/test_dyn_rnn.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/python/paddle/fluid/tests/unittests/test_dyn_rnn.py b/python/paddle/fluid/tests/unittests/test_dyn_rnn.py index 78f4669f7a..84fee8ace3 100644 --- a/python/paddle/fluid/tests/unittests/test_dyn_rnn.py +++ b/python/paddle/fluid/tests/unittests/test_dyn_rnn.py @@ -27,6 +27,8 @@ from paddle.fluid.layers.control_flow import array_to_lod_tensor from paddle.fluid.layers.control_flow import shrink_memory from fake_reader import fake_imdb_reader +numpy.random.seed(2020) + class TestDynamicRNN(unittest.TestCase): def setUp(self): @@ -144,6 +146,8 @@ class TestDynamicRNN(unittest.TestCase): def test_train_dynamic_rnn(self): main_program = fluid.Program() startup_program = fluid.Program() + main_program.random_seed = 10 + startup_program.random_seed = 10 with fluid.program_guard(main_program, startup_program): sentence = fluid.layers.data( name='word', shape=[1], dtype='int64', lod_level=1) @@ -196,6 +200,8 @@ class TestDynamicRNN(unittest.TestCase): main_program = fluid.Program() startup_program = fluid.Program() + main_program.random_seed = 10 + startup_program.random_seed = 10 with fluid.program_guard(main_program, startup_program): sentence = fluid.layers.data( name='word', shape=[1], dtype='int64', lod_level=2) @@ -249,6 +255,8 @@ class TestDynamicRNN(unittest.TestCase): hidden_size = 32 main_program = fluid.Program() startup_program = fluid.Program() + main_program.random_seed = 10 + startup_program.random_seed = 10 with fluid.program_guard(main_program, startup_program): sentence = fluid.layers.data( name='word', shape=[1], dtype='int64', lod_level=2) -- GitLab