From 17cd3c47e5e52be0aa6f3287222dfe8f740fc52c Mon Sep 17 00:00:00 2001
From: jingyong hou <houjingyong@gmail.com>
Date: Fri, 19 Nov 2021 17:18:58 +0800
Subject: [PATCH] update train.py

---
 kws/bin/train.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/kws/bin/train.py b/kws/bin/train.py
index df5e0fb..3497322 100644
--- a/kws/bin/train.py
+++ b/kws/bin/train.py
@@ -103,9 +103,9 @@ def main():
     logging.basicConfig(level=logging.DEBUG,
                         format='%(asctime)s %(levelname)s %(message)s')
     os.environ['CUDA_VISIBLE_DEVICES'] = str(args.gpu)
-    set_mannul_seed(args.gpu)
+
     # Set random seed
-    torch.manual_seed(777)
+    set_mannul_seed(args.seed)
     print(args)
     with open(args.config, 'r') as fin:
         configs = yaml.load(fin, Loader=yaml.FullLoader)