funasr/train_utils/load_pretrained_model.py
@@ -90,9 +90,11 @@ if dst_prefix == "" and (src_prefix + k) in src_state.keys(): k_src = src_prefix + k if not k_src.startswith("module."): print(f"init param, map: {k} from {k_src} in ckpt") elif k.startswith(dst_prefix) and k.replace(dst_prefix, src_prefix, 1) in src_state.keys(): k_src = k.replace(dst_prefix, src_prefix, 1) if not k_src.startswith("module."): print(f"init param, map: {k} from {k_src} in ckpt") if k_src in src_state.keys():
