|
|
|
@ -231,7 +231,7 @@ class TestImperativeResnet(unittest.TestCase):
|
|
|
|
seed = 90
|
|
|
|
seed = 90
|
|
|
|
|
|
|
|
|
|
|
|
batch_size = train_parameters["batch_size"]
|
|
|
|
batch_size = train_parameters["batch_size"]
|
|
|
|
batch_num = 50
|
|
|
|
batch_num = 20
|
|
|
|
with fluid.imperative.guard():
|
|
|
|
with fluid.imperative.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
|
|
|
|
|