[Cleanup][C-16] clean some random_seed for test (#61510)

Liyulingyue · web-flow · commit 537008d46bbe · 2024-02-03T19:50:37.000+08:00
diff --git a/test/dygraph_to_static/test_save_inference_model.py b/test/dygraph_to_static/test_save_inference_model.py
@@ -92,8 +92,7 @@ def tearDown(self):
     def test_save_inference_model(self):
         fc_size = 20
         x_data = np.random.random((fc_size, fc_size)).astype('float32')
-        base.default_startup_program().random_seed = SEED
-        base.default_main_program().random_seed = SEED
+        paddle.seed(SEED)
 
         x = paddle.to_tensor(x_data)
         layer = paddle.jit.to_static(SimpleFcLayer(fc_size))
@@ -139,8 +138,7 @@ def test_save_pylayer_model(self):
         x_data = np.random.random((fc_size, fc_size)).astype('float32')
         paddle.framework._set_expected_place(place)
 
-        base.default_startup_program().random_seed = SEED
-        base.default_main_program().random_seed = SEED
+        paddle.seed(SEED)
         x = paddle.to_tensor(x_data)
         layer = paddle.jit.to_static(SimplePyLayerNet(fc_size))
         adam = paddle.optimizer.SGD(
diff --git a/test/dygraph_to_static/test_seq2seq.py b/test/dygraph_to_static/test_seq2seq.py
@@ -52,8 +52,7 @@ def prepare_input(batch):
 
 def train(args, attn_model=False):
     with base.dygraph.guard(place):
-        paddle.static.default_startup_program().random_seed = 2020
-        paddle.static.default_main_program().random_seed = 2020
+        paddle.seed(2020)
 
         if attn_model:
             model = paddle.jit.to_static(
diff --git a/test/legacy_test/test_rnn_decode_api.py b/test/legacy_test/test_rnn_decode_api.py
@@ -177,8 +177,7 @@ def __init__(
             base.Program() if startup_program is None else startup_program
         )
         if seed is not None:
-            self.main_program.random_seed = seed
-            self.startup_program.random_seed = seed
+            paddle.seed(seed)
         self.build_program(model_cls, alg_cls, model_hparams, alg_hparams)
         self.executor = executor
 
diff --git a/test/legacy_test/test_static_save_load.py b/test/legacy_test/test_static_save_load.py
@@ -266,8 +266,7 @@ def test_ptb_rnn_cpu_float32(self):
         temp_dir = tempfile.TemporaryDirectory()
 
         with new_program_scope():
-            base.default_startup_program().random_seed = seed
-            base.default_main_program().random_seed = seed
+            paddle.seed(seed)
             ptb_model = PtbModel(
                 "ptb_model",
                 hidden_size=hidden_size,
@@ -397,8 +396,7 @@ def test_ptb_rnn_cpu_float32(self):
         temp_dir = tempfile.TemporaryDirectory()
 
         with new_program_scope():
-            base.default_startup_program().random_seed = seed
-            base.default_main_program().random_seed = seed
+            paddle.seed(seed)
             ptb_model = PtbModel(
                 "ptb_model",
                 hidden_size=hidden_size,
@@ -541,8 +539,7 @@ def test_ptb_rnn_cpu_float32(self):
         temp_dir = tempfile.TemporaryDirectory()
 
         with new_program_scope():
-            base.default_startup_program().random_seed = seed
-            base.default_main_program().random_seed = seed
+            paddle.seed(seed)
             ptb_model = PtbModel(
                 "ptb_model",
                 hidden_size=hidden_size,
@@ -670,8 +667,7 @@ def test_ptb_rnn_cpu_float32(self):
         temp_dir = tempfile.TemporaryDirectory()
 
         with new_program_scope():
-            base.default_startup_program().random_seed = seed
-            base.default_main_program().random_seed = seed
+            paddle.seed(seed)
             ptb_model = PtbModel(
                 "ptb_model",
                 hidden_size=hidden_size,
@@ -1001,8 +997,7 @@ def test_load_from_old_interface(self):
         batch_num = 200
 
         with new_program_scope():
-            base.default_startup_program().random_seed = seed
-            base.default_main_program().random_seed = seed
+            paddle.seed(seed)
             ptb_model = PtbModel(
                 "ptb_model",
                 hidden_size=hidden_size,
@@ -1144,8 +1139,7 @@ def test_load_from_old_interface_var_list(self):
         batch_num = 200
 
         with new_program_scope():
-            base.default_startup_program().random_seed = seed
-            base.default_main_program().random_seed = seed
+            paddle.seed(seed)
             ptb_model = PtbModel(
                 "ptb_model",
                 hidden_size=hidden_size,
@@ -1287,8 +1281,7 @@ def test_load_from_old_interface(self):
         temp_dir = tempfile.TemporaryDirectory()
 
         with new_program_scope():
-            base.default_startup_program().random_seed = seed
-            base.default_main_program().random_seed = seed
+            paddle.seed(seed)
             ptb_model = PtbModel(
                 "ptb_model",
                 hidden_size=hidden_size,
@@ -1479,8 +1472,7 @@ def test_ptb_rnn_cpu_float32(self):
         batch_num = 200
 
         with new_program_scope():
-            base.default_startup_program().random_seed = seed
-            base.default_main_program().random_seed = seed
+            paddle.seed(seed)
             ptb_model = PtbModel(
                 "ptb_model",
                 hidden_size=hidden_size,
@@ -1655,8 +1647,7 @@ def test_ptb_rnn_cpu_float32(self):
         temp_dir = tempfile.TemporaryDirectory()
 
         with new_program_scope():
-            base.default_startup_program().random_seed = seed
-            base.default_main_program().random_seed = seed
+            paddle.seed(seed)
             ptb_model = PtbModel(
                 "ptb_model",
                 hidden_size=hidden_size,
diff --git a/test/quantization/test_quantization_scale_pass.py b/test/quantization/test_quantization_scale_pass.py
@@ -75,8 +75,7 @@ def quantization_scale(
         for_ci=False,
     ):
         def build_program(main, startup, is_test):
-            main.random_seed = seed
-            startup.random_seed = seed
+            paddle.seed(2023)
             with paddle.utils.unique_name.guard():
                 with paddle.static.program_guard(main, startup):
                     img = paddle.static.data(

Original file line number	Diff line number	Diff line change
`@@ -177,8 +177,7 @@ def __init__(`
`177`	`177`	`base.Program() if startup_program is None else startup_program`
`178`	`178`	`)`
`179`	`179`	`if seed is not None:`
`180`		`- self.main_program.random_seed = seed`
`181`		`- self.startup_program.random_seed = seed`
	`180`	`+ paddle.seed(seed)`
`182`	`181`	`self.build_program(model_cls, alg_cls, model_hparams, alg_hparams)`
`183`	`182`	`self.executor = executor`
`184`	`183`