From d6d3facaa946620ba872691a2f8084fa7a4e051c Mon Sep 17 00:00:00 2001
From: Bobholamovic <bob1998425@hotmail.com>
Date: Thu, 1 Oct 2020 16:44:14 +0800
Subject: [PATCH] Add cuda manual seed

---
 src/train.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/train.py b/src/train.py
index c278954..84b7769 100644
--- a/src/train.py
+++ b/src/train.py
@@ -142,6 +142,7 @@ def main():
     random.seed(RNG_SEED)
     np.random.seed(RNG_SEED)
     torch.manual_seed(RNG_SEED)
+    torch.cuda.manual_seed(RNG_SEED)
 
     cudnn.deterministic = True
     cudnn.benchmark = False
-- 
GitLab