diff --git a/fluid/PaddleCV/HiNAS_models/nn_paddle.py b/fluid/PaddleCV/HiNAS_models/nn_paddle.py index 8cb30673cef3e2cde48b22c7e950594dfa0ecd56..c91d3d800bee0fdda689c6cd25a689c30a3b6681 100755 --- a/fluid/PaddleCV/HiNAS_models/nn_paddle.py +++ b/fluid/PaddleCV/HiNAS_models/nn_paddle.py @@ -94,6 +94,14 @@ class Model(object): return [avg_cost, accuracy] def run(self): + + if FLAGS.enable_ce: + import random + SEED = 102 + random.seed(SEED) + np.random.seed(SEED) + fluid.default_main_program().random_seed = SEED + fluid.default_startup_program().random_seed = SEED train_files = reader.train10() test_files = reader.test10() diff --git a/fluid/PaddleCV/HiNAS_models/reader.py b/fluid/PaddleCV/HiNAS_models/reader.py index 9448af4e7d578e480f136ba77167136074dd6203..9b28a2ee96c39f0a95030f7b695c67809d890669 100755 --- a/fluid/PaddleCV/HiNAS_models/reader.py +++ b/fluid/PaddleCV/HiNAS_models/reader.py @@ -82,8 +82,6 @@ def preprocess(sample, is_training): img = (img_float - mean) / std if is_training and FLAGS.cutout: - if FLAGS.enable_ce: - np.random.seed(0) center = np.random.randint(image_size, size=2) offset_width = max(0, center[0] - half_length) offset_height = max(0, center[1] - half_length)