From 82a07e2f6ec60aa25a3931e9ee0d99ead642484a Mon Sep 17 00:00:00 2001
From: Kyle He <599316527@users.noreply.github.com>
Date: 星期四, 14 八月 2025 16:07:54 +0800
Subject: [PATCH] fix: funasr_onnx pass run_options to ort (#2632)
---
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