Merge branch 'master' into shuai-dev-cli

wenet-e2e · Nov 13, 2023 · 35fef8c · 35fef8c
2 parents 8374f91 + dd02196
commit 35fef8c
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/wespeaker/cli/speaker.py b/wespeaker/cli/speaker.py
@@ -183,6 +183,11 @@ def get_args():
     parser.add_argument('--vad',
                         action='store_true',
                         help='whether to do VAD or not')
+    parser.add_argument('-g',
+                        '--gpu',
+                        type=int,
+                        default=-1,
+                        help='which gpu to use (number <0 means using cpu)')
     parser.add_argument('--output_file',
                         help='output file to save speaker embedding')
     args = parser.parse_args()