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

fix_ema
wopeizl 6 years ago committed by GitHub
parent e6b5a3f092
commit 3bd14263f5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

@ -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

Loading…
Cancel
Save