From 8672352ecde80a86609fe01195b398ebe77f0ed1 Mon Sep 17 00:00:00 2001
From: aky15 <ankeyu.aky@11.17.44.249>
Date: 星期一, 17 四月 2023 16:09:23 +0800
Subject: [PATCH] merge many functions
---
funasr/models/decoder/rnnt_decoder.py | 3 +--
1 files changed, 1 insertions(+), 2 deletions(-)
diff --git a/funasr/models/rnnt_predictor/rnn_decoder.py b/funasr/models/decoder/rnnt_decoder.py
similarity index 98%
rename from funasr/models/rnnt_predictor/rnn_decoder.py
rename to funasr/models/decoder/rnnt_decoder.py
index 0df6fc7..5401ab2 100644
--- a/funasr/models/rnnt_predictor/rnn_decoder.py
+++ b/funasr/models/decoder/rnnt_decoder.py
@@ -6,10 +6,9 @@
from typeguard import check_argument_types
from funasr.modules.beam_search.beam_search_transducer import Hypothesis
-from funasr.models.rnnt_predictor.abs_decoder import AbsDecoder
from funasr.models.specaug.specaug import SpecAug
-class RNNDecoder(AbsDecoder):
+class RNNTDecoder(torch.nn.Module):
"""RNN decoder module.
Args:
--
Gitblit v1.9.1