diff --git a/python/paddle/fluid/tests/unittests/test_imperative_resnet.py b/python/paddle/fluid/tests/unittests/test_imperative_resnet.py index d9ef08b3c491b24323bb1469165ed5482737013a..a0cfb27d47fd2320902e35df8fb696a71d244084 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 77e6fc273423299453be2fde5e6e4cde468f5d0e..74560535074f2550429de79385415162987dab7d 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 3f3f92cde57c80fa4ba3d2f1389cc47efd74ca5b..ae6a73904d6b5d62d56355a4f2b13426a0d8e7f0 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