From 1596f6f414f6f41da66506debb1dff19fffeb3ec Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期一, 24 六月 2024 11:55:17 +0800
Subject: [PATCH] fixbug hotwords
---
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