From 5241491e951572eb83778fa36eacdcfadf82af37 Mon Sep 17 00:00:00 2001 From: Binbin Zhang Date: Mon, 6 Dec 2021 11:38:41 +0800 Subject: [PATCH] [kws] fix seed type (#26) --- kws/bin/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/kws/bin/train.py b/kws/bin/train.py index 6ba902b..f6fe33d 100644 --- 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',