[kws] fix seed type (#26)

2021-12-06 11:38:41 +08:00 · 2021-12-06 11:38:41 +08:00 · 5241491e95
commit 5241491e95
parent a5a54782cc
1 changed files with 1 additions and 1 deletions
--- a/kws/bin/train.py
+++ b/kws/bin/train.py
@ -43,7 +43,7 @@ def get_args():
                        default=-1,
                        help='gpu id for this local rank, -1 for cpu')
    parser.add_argument('--model_dir', required=True, help='save model dir')
-    parser.add_argument('--seed', dest='seed', default=777, help='random seed')
+    parser.add_argument('--seed', type=int, default=777, help='random seed')
    parser.add_argument('--checkpoint', help='checkpoint model')
    parser.add_argument('--tensorboard_dir',
                        default='tensorboard',