未验证 提交 3bd14263 编写于 作者: W wopeizl 提交者: GitHub

decrease the train loop number to avoid run too long to fail the ci process test=develop (#17567)

上级 e6b5a3f0
...@@ -231,7 +231,8 @@ class TestDygraphResnet(unittest.TestCase): ...@@ -231,7 +231,8 @@ class TestDygraphResnet(unittest.TestCase):
seed = 90 seed = 90
batch_size = train_parameters["batch_size"] batch_size = train_parameters["batch_size"]
batch_num = 20 batch_num = 10
with fluid.dygraph.guard(): with fluid.dygraph.guard():
fluid.default_startup_program().random_seed = seed fluid.default_startup_program().random_seed = seed
fluid.default_main_program().random_seed = seed fluid.default_main_program().random_seed = seed
......
...@@ -71,7 +71,7 @@ class TestDygraphResnetSortGradient(unittest.TestCase): ...@@ -71,7 +71,7 @@ class TestDygraphResnetSortGradient(unittest.TestCase):
seed = 90 seed = 90
batch_size = train_parameters["batch_size"] batch_size = train_parameters["batch_size"]
batch_num = 20 batch_num = 10
with fluid.dygraph.guard(): with fluid.dygraph.guard():
fluid.default_startup_program().random_seed = seed fluid.default_startup_program().random_seed = seed
fluid.default_main_program().random_seed = seed fluid.default_main_program().random_seed = seed
......
...@@ -315,7 +315,7 @@ class TestImperativeResneXt(unittest.TestCase): ...@@ -315,7 +315,7 @@ class TestImperativeResneXt(unittest.TestCase):
seed = 90 seed = 90
batch_size = train_parameters["batch_size"] batch_size = train_parameters["batch_size"]
batch_num = 2 batch_num = 1
epoch_num = 1 epoch_num = 1
with fluid.dygraph.guard(): with fluid.dygraph.guard():
fluid.default_startup_program().random_seed = seed fluid.default_startup_program().random_seed = seed
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册