From a8591060d3889cd7a72841fa32a7ee64b49db1d2 Mon Sep 17 00:00:00 2001
From: msgk <zxr935867802@outlook.com>
Date: 星期五, 14 二月 2025 14:16:51 +0800
Subject: [PATCH] fix(spk): 修复 speaker embedding 集群后的重新排序问题
---
funasr/train_utils/initialize.py | 6 +-----
1 files changed, 1 insertions(+), 5 deletions(-)
diff --git a/funasr/train_utils/initialize.py b/funasr/train_utils/initialize.py
index ee94850..667585a 100644
--- a/funasr/train_utils/initialize.py
+++ b/funasr/train_utils/initialize.py
@@ -19,7 +19,6 @@
init: Method of initialization.
"""
-
# weight init
for p in model.parameters():
if p.dim() > 1:
@@ -40,9 +39,7 @@
# reset some modules with default init
for m in model.modules():
- if isinstance(
- m, (torch.nn.Embedding, torch.nn.LayerNorm, torch.nn.GroupNorm)
- ):
+ if isinstance(m, (torch.nn.Embedding, torch.nn.LayerNorm, torch.nn.GroupNorm)):
m.reset_parameters()
if hasattr(m, "espnet_initialization_fn"):
m.espnet_initialization_fn()
@@ -56,4 +53,3 @@
model.frontend, "reload_pretrained_parameters", None
):
model.frontend.reload_pretrained_parameters()
-
--
Gitblit v1.9.1