inference

游雁

2023-05-16 d105ce0d6b63bcd14edeb426fbc0acf593296be3

 funasr/bin/asr_train_paraformer.py

@@ -1,4 +1,7 @@
# -*- encoding: utf-8 -*-
#!/usr/bin/env python3
# Copyright FunASR (https://github.com/alibaba-damo-academy/FunASR). All Rights Reserved.
#  MIT License  (https://opensource.org/licenses/MIT)

import os

@@ -9,6 +12,12 @@
def parse_args():
    parser = ASRTask.get_parser()
    parser.add_argument(
        "--mode",
        type=str,
        default="asr",
        help="mode",
    )
    parser.add_argument(
        "--gpu_id",
        type=int,
        default=0,