update train.py

2021-11-19 17:18:58 +08:00 · 2021-11-19 17:18:58 +08:00 · 17cd3c47e5
commit 17cd3c47e5
parent 8cd12edfed
1 changed files with 2 additions and 2 deletions
--- 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)