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

update

Browse Source
This commit is contained in:
游雁 2024-03-22 20:35:58 +08:00
parent 81fb78286f
commit 334422ead3
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
funasr/train_utils/trainer.py
View File

@ -198,6 +198,8 @@ class Trainer:
for k in dst_state.keys():
if not k.startswith("module.") and "module."+k in src_state.keys():
k_ddp = "module."+k
elif k.startswith("module.") and "module."+k not in src_state.keys():
k_ddp = k.replace("module.", "", 1)
else:
k_ddp = k
if k_ddp in src_state.keys():