funasr2

游雁

2023-12-21 c8bae0ec85eee25d66de6b1e4502eff74d750b24

 funasr/train_utils/load_pretrained_model.py

@@ -120,6 +120,7 @@
    if ignore_init_mismatch:
        src_state = filter_state_dict(dst_state, src_state)

    # logging.info("Loaded src_state keys: {}".format(src_state.keys()))
    logging.debug("Loaded src_state keys: {}".format(src_state.keys()))
    logging.debug("Loaded dst_state keys: {}".format(dst_state.keys()))
    dst_state.update(src_state)
    obj.load_state_dict(dst_state)