fix random seed

zdaiot · Oct 25, 2019 · 2af4de5 · 2af4de5
1 parent c8874ef
commit 2af4de5
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/train_classify.py b/train_classify.py
@@ -52,7 +52,8 @@ def __init__(self, config, fold):
 
         self.max_accuracy_valid = 0
         # 设置随机种子，注意交叉验证部分划分训练集和验证集的时候，要保持种子固定
-        self.seed = 1570421136
+        self.seed = int(time.time())
+        # self.seed = 1570421136
         seed_torch(self.seed)
         with open(self.model_path + '/'+ TIMESTAMP + '.pkl','wb') as f:
             pickle.dump({'seed': self.seed}, f, -1)

diff --git a/train_segment.py b/train_segment.py
@@ -58,7 +58,8 @@ def __init__(self, config, fold):
         self.max_dice_valid = 0
 
         # 设置随机种子，注意交叉验证部分划分训练集和验证集的时候，要保持种子固定
-        self.seed = 1570421136
+        self.seed = int(time.time())
+        # self.seed = 1570421136
         seed_torch(self.seed)
         with open(self.model_path + '/'+ TIMESTAMP + '.pkl','wb') as f:
             pickle.dump({'seed': self.seed}, f, -1)