diff --git a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_mobile_net.py b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_mobile_net.py index 49002ed57da23417a7d497dbadab3c178b791425..33b5860d7fd1fcdfa13743fc8d1edce6d00e77a4 100644 --- a/python/paddle/fluid/tests/unittests/dygraph_to_static/test_mobile_net.py +++ b/python/paddle/fluid/tests/unittests/dygraph_to_static/test_mobile_net.py @@ -407,12 +407,14 @@ def create_optimizer(args, parameter_list): return optimizer -def fake_data_reader(batch_size, lable_size): +def fake_data_reader(batch_size, label_size): + local_random = np.random.RandomState(SEED) + def reader(): batch_data = [] while True: - img = np.random.random([3, 224, 224]).astype('float32') - label = np.random.randint(0, lable_size, [1]).astype('int64') + img = local_random.random_sample([3, 224, 224]).astype('float32') + label = local_random.randint(0, label_size, [1]).astype('int64') batch_data.append([img, label]) if len(batch_data) == batch_size: yield batch_data