diff --git a/funasr/train_utils/load_pretrained_model.py b/funasr/train_utils/load_pretrained_model.py index 105e0bd18..8ed613caf 100644 --- a/funasr/train_utils/load_pretrained_model.py +++ b/funasr/train_utils/load_pretrained_model.py @@ -96,7 +96,7 @@ def load_pretrained_model( dst_state[k] = src_state[k_src] else: - print(f"Warning, miss key in ckpt: {k}") + print(f"Warning, miss key in ckpt: {k}, {path}") flag = obj.load_state_dict(dst_state, strict=True) logging.info(f"Loading ckpt: {path}, status: {flag}")