From 518465d0898dd1172268e901e990374f6deb5461 Mon Sep 17 00:00:00 2001
From: speech_asr <wangjiaming.wjm@alibaba-inc.com>
Date: 星期四, 20 四月 2023 16:07:01 +0800
Subject: [PATCH] update
---
funasr/bin/train.py | 1 +
funasr/utils/build_asr_model.py | 2 ++
2 files changed, 3 insertions(+), 0 deletions(-)
diff --git a/funasr/bin/train.py b/funasr/bin/train.py
index cbbf1fa..8acd37c 100644
--- a/funasr/bin/train.py
+++ b/funasr/bin/train.py
@@ -11,6 +11,7 @@
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
diff --git a/funasr/utils/build_asr_model.py b/funasr/utils/build_asr_model.py
index 4908cd4..f333969 100644
--- a/funasr/utils/build_asr_model.py
+++ b/funasr/utils/build_asr_model.py
@@ -349,3 +349,5 @@
if args.init is not None:
initialize(model, args.init)
+
+ return model
\ No newline at end of file
--
Gitblit v1.9.1