funasr/bin/asr_train.py
@@ -34,6 +34,10 @@ from funasr.tasks.asr import ASRTask if args.mode == "paraformer": from funasr.tasks.asr import ASRTaskParaformer as ASRTask if args.mode == "uniasr": from funasr.tasks.asr import ASRTaskUniASR as ASRTask if args.mode == "rnnt": from funasr.tasks.asr import ASRTransducerTask as ASRTask ASRTask.main(args=args, cmd=cmd) @@ -60,3 +64,4 @@ args.batch_bins = args.batch_bins * args.ngpu main(args=args)