decoding

游雁

2024-06-13 a51234d426f540cb6e3998d4063c52dada016c2a

decoding

 funasr/train_utils/load_pretrained_model.py

@@ -96,7 +96,7 @@
                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}")