未验证 提交 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):
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
......
......@@ -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
......
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册