From 0efc87352ce7d3903dbdedbfa5d01ca5e1cb19e7 Mon Sep 17 00:00:00 2001
From: Shi Xian <40013335+R1ckShi@users.noreply.github.com>
Date: 星期四, 05 十二月 2024 15:15:38 +0800
Subject: [PATCH] Merge pull request #2267 from modelscope/dev_sx2
---
funasr/frontends/fused.py | 14 ++++----------
1 files changed, 4 insertions(+), 10 deletions(-)
diff --git a/funasr/frontends/fused.py b/funasr/frontends/fused.py
index 24f73f4..0fa7639 100644
--- a/funasr/frontends/fused.py
+++ b/funasr/frontends/fused.py
@@ -7,14 +7,10 @@
class FusedFrontends(nn.Module):
- def __init__(
- self, frontends=None, align_method="linear_projection", proj_dim=100, fs=16000
- ):
+ def __init__(self, frontends=None, align_method="linear_projection", proj_dim=100, fs=16000):
super().__init__()
- self.align_method = (
- align_method # fusing method : linear_projection only for now
- )
+ self.align_method = align_method # fusing method : linear_projection only for now
self.proj_dim = proj_dim # dim of the projection done on each frontend
self.frontends = [] # list of the frontends to combine
@@ -109,9 +105,7 @@
input_feats, feats_lens = frontend.forward(input, input_lengths)
self.feats.append([input_feats, feats_lens])
- if (
- self.align_method == "linear_projection"
- ): # TODO(Dan): to add other align methods
+ if self.align_method == "linear_projection": # TODO(Dan): to add other align methods
# first step : projections
self.feats_proj = []
@@ -141,4 +135,4 @@
else:
raise NotImplementedError
- return input_feats, feats_lens
\ No newline at end of file
+ return input_feats, feats_lens
--
Gitblit v1.9.1