From 3df109adfccedeb134dea4ba2ea9a2da89872048 Mon Sep 17 00:00:00 2001
From: Isuxiz Slidder <48672727+Isuxiz@users.noreply.github.com>
Date: 星期一, 31 三月 2025 17:51:52 +0800
Subject: [PATCH] Update model.py to fix "IndexError: index 1 is out of bounds for dimension 1 with size 0" (#2454)
---
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