雾聪
2024-07-01 31bf3a88a09f6e7c895224f4dc75e8f2c138d5c8
funasr/train_utils/load_pretrained_model.py
@@ -96,7 +96,7 @@
                dst_state[k] = src_state[k_src]
        else:
            logging.info(f"Warning, miss key in ckpt: {k}, mapped: {k_src}")
            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}")