From 3bd14263f53ea61b97ea6d86b276cf291000e4af Mon Sep 17 00:00:00 2001 From: wopeizl Date: Thu, 23 May 2019 10:42:33 +0800 Subject: [PATCH] decrease the train loop number to avoid run too long to fail the ci process test=develop (#17567) --- python/paddle/fluid/tests/unittests/test_imperative_resnet.py | 3 ++- .../tests/unittests/test_imperative_resnet_sorted_gradient.py | 2 +- .../paddle/fluid/tests/unittests/test_imperative_se_resnext.py | 2 +- 3 files changed, 4 insertions(+), 3 deletions(-) diff --git a/python/paddle/fluid/tests/unittests/test_imperative_resnet.py b/python/paddle/fluid/tests/unittests/test_imperative_resnet.py index d9ef08b3c49..a0cfb27d47f 100644 --- a/python/paddle/fluid/tests/unittests/test_imperative_resnet.py +++ b/python/paddle/fluid/tests/unittests/test_imperative_resnet.py @@ -231,7 +231,8 @@ class TestDygraphResnet(unittest.TestCase): seed = 90 batch_size = train_parameters["batch_size"] - batch_num = 20 + batch_num = 10 + with fluid.dygraph.guard(): fluid.default_startup_program().random_seed = seed fluid.default_main_program().random_seed = seed diff --git a/python/paddle/fluid/tests/unittests/test_imperative_resnet_sorted_gradient.py b/python/paddle/fluid/tests/unittests/test_imperative_resnet_sorted_gradient.py index 77e6fc27342..74560535074 100644 --- a/python/paddle/fluid/tests/unittests/test_imperative_resnet_sorted_gradient.py +++ b/python/paddle/fluid/tests/unittests/test_imperative_resnet_sorted_gradient.py @@ -71,7 +71,7 @@ class TestDygraphResnetSortGradient(unittest.TestCase): seed = 90 batch_size = train_parameters["batch_size"] - batch_num = 20 + batch_num = 10 with fluid.dygraph.guard(): fluid.default_startup_program().random_seed = seed fluid.default_main_program().random_seed = seed diff --git a/python/paddle/fluid/tests/unittests/test_imperative_se_resnext.py b/python/paddle/fluid/tests/unittests/test_imperative_se_resnext.py index 3f3f92cde57..ae6a73904d6 100644 --- a/python/paddle/fluid/tests/unittests/test_imperative_se_resnext.py +++ b/python/paddle/fluid/tests/unittests/test_imperative_se_resnext.py @@ -315,7 +315,7 @@ class TestImperativeResneXt(unittest.TestCase): seed = 90 batch_size = train_parameters["batch_size"] - batch_num = 2 + batch_num = 1 epoch_num = 1 with fluid.dygraph.guard(): fluid.default_startup_program().random_seed = seed -- GitLab