This commit is contained in:
speech_asr 2023-04-20 17:37:31 +08:00
parent fac8dbd821
commit 77e4b6899b

View File

@ -13,6 +13,8 @@ def build_model(args):
model = build_lm_model(args)
elif args.task_name == "punc":
model = build_punc_model(args)
elif args.task_name == "vad":
model = build_vad_model(args)
else:
raise NotImplementedError("Not supported task: {}".format(args.task_name))