From ec42889511b121230e97bbcdf05f4d517f95d7ba Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期五, 23 二月 2024 15:09:35 +0800
Subject: [PATCH] update
---
funasr/train_utils/load_pretrained_model.py | 2 +-
1 files changed, 1 insertions(+), 1 deletions(-)
diff --git a/funasr/train_utils/load_pretrained_model.py b/funasr/train_utils/load_pretrained_model.py
index 8493bf5..03a6ff5 100644
--- a/funasr/train_utils/load_pretrained_model.py
+++ b/funasr/train_utils/load_pretrained_model.py
@@ -118,7 +118,7 @@
else:
print(f"Warning, miss key in ckpt: {k}, mapped: {k_ddp}")
- flag = obj.load_state_dict(dst_state, strict=True)
+ flag = obj.load_state_dict(dst_state, strict=False)
# print(flag)
# def load_pretrained_model(
--
Gitblit v1.9.1