游雁
2024-03-22 334422ead333700cc9320e841eac3057e94f6437
update
1个文件已修改
2 ■■■■■ 已修改文件
funasr/train_utils/trainer.py 2 ●●●●● 补丁 | 查看 | 原始文档 | 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():