This commit is contained in:
speech_asr 2023-04-20 16:07:01 +08:00
parent a29166b9a0
commit 518465d089
2 changed files with 3 additions and 0 deletions

View File

@ -11,6 +11,7 @@ from funasr.torch_utils.set_all_random_seed import set_all_random_seed
from funasr.utils import config_argparse
from funasr.utils.build_dataloader import build_dataloader
from funasr.utils.build_distributed import build_distributed
from funasr.utils.build_model import build_model
from funasr.utils.build_optimizer import build_optimizer
from funasr.utils.build_scheduler import build_scheduler
from funasr.utils.prepare_data import prepare_data

View File

@ -349,3 +349,5 @@ def build_asr_model(args):
if args.init is not None:
initialize(model, args.init)
return model