| funasr/train_utils/trainer.py | ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史 |
funasr/train_utils/trainer.py
@@ -198,6 +198,8 @@ 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():