From 5115a066c971b2380c98f9be168e14869053eda1 Mon Sep 17 00:00:00 2001
From: majic31 <majic31@163.com>
Date: 星期四, 14 八月 2025 16:08:44 +0800
Subject: [PATCH] fix #2587: Resolve VAD multithreading issue (#2613)
---
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