From e24dbdc496debec225414d4d2c760f5775e64f2a Mon Sep 17 00:00:00 2001
From: 天地 <tiandiweizun@gmail.com>
Date: 星期三, 26 三月 2025 13:44:41 +0800
Subject: [PATCH] 感觉应该从文件读取更合适，因为上面判断了文件存在，且可以读取，如果本身是文本的话，下面也会有逻辑进行处理 (#2452)
---
funasr/models/sond/sv_decoder.py | 20 ++++++++++++--------
1 files changed, 12 insertions(+), 8 deletions(-)
diff --git a/funasr/models/sond/sv_decoder.py b/funasr/models/sond/sv_decoder.py
index ee6b0e5..7d8484d 100644
--- a/funasr/models/sond/sv_decoder.py
+++ b/funasr/models/sond/sv_decoder.py
@@ -5,19 +5,23 @@
class DenseDecoder(AbsDecoder):
def __init__(
- self,
- vocab_size,
- encoder_output_size,
- num_nodes_resnet1: int = 256,
- num_nodes_last_layer: int = 256,
- batchnorm_momentum: float = 0.5,
+ self,
+ vocab_size,
+ encoder_output_size,
+ num_nodes_resnet1: int = 256,
+ num_nodes_last_layer: int = 256,
+ batchnorm_momentum: float = 0.5,
):
super(DenseDecoder, self).__init__()
self.resnet1_dense = torch.nn.Linear(encoder_output_size, num_nodes_resnet1)
- self.resnet1_bn = torch.nn.BatchNorm1d(num_nodes_resnet1, eps=1e-3, momentum=batchnorm_momentum)
+ self.resnet1_bn = torch.nn.BatchNorm1d(
+ num_nodes_resnet1, eps=1e-3, momentum=batchnorm_momentum
+ )
self.resnet2_dense = torch.nn.Linear(num_nodes_resnet1, num_nodes_last_layer)
- self.resnet2_bn = torch.nn.BatchNorm1d(num_nodes_last_layer, eps=1e-3, momentum=batchnorm_momentum)
+ self.resnet2_bn = torch.nn.BatchNorm1d(
+ num_nodes_last_layer, eps=1e-3, momentum=batchnorm_momentum
+ )
self.output_dense = torch.nn.Linear(num_nodes_last_layer, vocab_size, bias=False)
--
Gitblit v1.9.1