AI/wekws
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[kws] fix seed type (#26)

Browse Source
This commit is contained in:
Binbin Zhang 2021-12-06 11:38:41 +08:00 committed by GitHub
parent a5a54782cc
commit 5241491e95
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
kws/bin/train.py
View File

@ -43,7 +43,7 @@ def get_args():
default=-1, default=-1,
help='gpu id for this local rank, -1 for cpu') help='gpu id for this local rank, -1 for cpu')
parser.add_argument('--model_dir', required=True, help='save model dir') 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('--checkpoint', help='checkpoint model')
parser.add_argument('--tensorboard_dir', parser.add_argument('--tensorboard_dir',
default='tensorboard', default='tensorboard',