From 7662bd4965766785a16e1ddbba289d27ea780a79 Mon Sep 17 00:00:00 2001
From: wangzhen38 <wangzhen38@baidu.com>
Date: Wed, 13 Jul 2022 07:22:24 +0000
Subject: [PATCH] [SEED FIX]fix seed of mind

---
 models/recall/mind/mind_reader.py            | 6 +++---
 test_tipc/configs/dnn/train_infer_python.txt | 1 +
 2 files changed, 4 insertions(+), 3 deletions(-)

diff --git a/models/recall/mind/mind_reader.py b/models/recall/mind/mind_reader.py
index 18be7d2b6..5771ecd25 100644
--- a/models/recall/mind/mind_reader.py
+++ b/models/recall/mind/mind_reader.py
@@ -17,7 +17,7 @@
 from paddle.io import IterableDataset
 import random
 
-random.seed(12345)
+#random.seed(12345)
 
 
 class RecDataset(IterableDataset):
@@ -53,7 +53,7 @@ def init(self):
         self.items = list(self.items)
 
     def __iter__(self):
-        # random.seed(12345)
+        random.seed(12345)
         while True:
             user_id_list = random.sample(self.users, self.batch_size)
             if self.count >= self.batches_per_epoch * self.batch_size:
@@ -63,7 +63,7 @@ def __iter__(self):
                 item_list = self.graph[user_id]
                 if len(item_list) <= 4:
                     continue
-                # random.seed(12345)
+                random.seed(12345)
                 k = random.choice(range(4, len(item_list)))
                 item_id = item_list[k]
 
diff --git a/test_tipc/configs/dnn/train_infer_python.txt b/test_tipc/configs/dnn/train_infer_python.txt
index 08fc02e68..20d07d69e 100755
--- a/test_tipc/configs/dnn/train_infer_python.txt
+++ b/test_tipc/configs/dnn/train_infer_python.txt
@@ -55,3 +55,4 @@ epoch:50
 --profiler_options="batch_range=[10,20];state=GPU;tracer_option=Default;profile_path=model.profile"
 run_mode:PSGPU
 fp_items:null
+device_num:N1C8