speech_recognition/FunASR
1
0
Fork 0
mirror of https://github.com/modelscope/FunASR synced 2025-09-15 14:48:36 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

decoding

Browse Source
This commit is contained in:
游雁 2024-06-12 14:49:24 +08:00
parent 9ec4f8c629
commit a56980a26f
1 changed files with 6 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
funasr/train_utils/load_pretrained_model.py
View File

@ -52,12 +52,11 @@ def load_pretrained_model(
if excludes is not None: if excludes is not None:
if isinstance(excludes, str): if isinstance(excludes, str):
excludes = excludes.split(",") excludes = excludes.split(",")
if not isinstance(excludes, (tuple, list)):
excludes = [excludes]
logging.info(f"excludes: {excludes}") logging.info(f"excludes: {excludes}")
for k in dst_state.keys(): for k in dst_state.keys():
if excludes is not None:
for k_ex in excludes: for k_ex in excludes:
if k.startswith(k_ex): if k.startswith(k_ex):
logging.info(f"key: {{k}} matching: {k_ex}, excluded") logging.info(f"key: {{k}} matching: {k_ex}, excluded")