shixian.shi
2023-05-04 8d169f9cd98f4b2a4917e0ced2638856e3410277
update
1个文件已修改
2 ■■■ 已修改文件
funasr/torch_utils/load_pretrained_model.py 2 ●●● 补丁 | 查看 | 原始文档 | blame | 历史
funasr/torch_utils/load_pretrained_model.py
@@ -120,6 +120,6 @@
    if ignore_init_mismatch:
        src_state = filter_state_dict(dst_state, src_state)
    logging.info("Loaded src_state keys: {}".format(src_state.keys()))
    # logging.info("Loaded src_state keys: {}".format(src_state.keys()))
    dst_state.update(src_state)
    obj.load_state_dict(dst_state)