From 334422ead333700cc9320e841eac3057e94f6437 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期五, 22 三月 2024 20:35:58 +0800
Subject: [PATCH] update
---
funasr/train_utils/trainer.py | 2 ++
1 files changed, 2 insertions(+), 0 deletions(-)
diff --git a/funasr/train_utils/trainer.py b/funasr/train_utils/trainer.py
index c443c6f..49b9fbc 100644
--- a/funasr/train_utils/trainer.py
+++ b/funasr/train_utils/trainer.py
@@ -198,6 +198,8 @@
for k in dst_state.keys():
if not k.startswith("module.") and "module."+k in src_state.keys():
k_ddp = "module."+k
+ elif k.startswith("module.") and "module."+k not in src_state.keys():
+ k_ddp = k.replace("module.", "", 1)
else:
k_ddp = k
if k_ddp in src_state.keys():
--
Gitblit v1.9.1