From 4a99a0ac273956a7f8e6608e71aafbb5202fcca8 Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期六, 11 五月 2024 21:55:14 +0800
Subject: [PATCH] sensevoice sanm
---
funasr/models/sense_voice/model.py | 12 ++++--------
1 files changed, 4 insertions(+), 8 deletions(-)
diff --git a/funasr/models/sense_voice/model.py b/funasr/models/sense_voice/model.py
index a633a8d..127d5a0 100644
--- a/funasr/models/sense_voice/model.py
+++ b/funasr/models/sense_voice/model.py
@@ -1005,9 +1005,7 @@
if specaug is not None:
specaug_class = tables.specaug_classes.get(specaug)
specaug = specaug_class(**specaug_conf)
- if normalize is not None:
- normalize_class = tables.normalize_classes.get(normalize)
- normalize = normalize_class(**normalize_conf)
+
encoder_class = tables.encoder_classes.get(encoder)
encoder = encoder_class(input_size=input_size, **encoder_conf)
encoder_output_size = encoder.output_size()
@@ -1026,7 +1024,7 @@
self.ignore_id = ignore_id
self.specaug = specaug
- self.normalize = normalize
+
self.encoder = encoder
self.decoder = decoder
@@ -1040,12 +1038,9 @@
self.error_calculator = None
- self.share_embedding = share_embedding
- if self.share_embedding:
- self.decoder.embed = None
-
self.length_normalized_loss = length_normalized_loss
self.beam_search = None
+ self.activation_checkpoint = kwargs.get("activation_checkpoint", False)
def forward(
self,
@@ -1139,6 +1134,7 @@
stats = {}
# 1. Forward decoder
+ ys_pad[ys_pad == -1] = 0
decoder_out = self.decoder(encoder_out, encoder_out_lens, ys_pad, ys_pad_lens)
if isinstance(decoder_out, (list, tuple)):
decoder_out = decoder_out[0]
--
Gitblit v1.9.1