train

游雁

2023-05-16 d5d7363da1f56c6932cba2901cc4b9d6f130a069

 funasr/bin/asr_train.py

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


def main(args=None, cmd=None):
    
    # for ASR Training
    if args.mode == "asr":
        from funasr.tasks.asr import ASRTask
    if args.mode == "paraformer":
        from funasr.tasks.asr import ASRTaskParaformer as ASRTask

    ASRTask.main(args=args, cmd=cmd)