fix seed

PaddlePaddle · Jan 15, 2024 · a3d8dcf · a3d8dcf
1 parent 523dcbe
commit a3d8dcf
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/python/paddle/nn/functional/common.py b/python/paddle/nn/functional/common.py
@@ -1136,6 +1136,10 @@ def dropout(
         if in_dynamic_or_pir_mode():
             if paddle.static.default_main_program().random_seed != 0:
                 seed = paddle.static.default_main_program().random_seed
+            if paddle.common_ops_import.default_main_program().random_seed != 0:
+                seed = (
+                    paddle.common_ops_import.default_main_program().random_seed
+                )
 
             out = _C_ops.dropout(
                 x,

diff --git a/python/paddle/pir/program_patch.py b/python/paddle/pir/program_patch.py
@@ -38,7 +38,6 @@ def _lr_schedule_guard(self, is_with_opt=False):
     program_attrs = {
         "global_seed": global_seed,
         "_seed": global_prog_seed,
-        "random_seed": None,
         "_lr_schedule_guard": _lr_schedule_guard,
     }