From 9b4e9cc8a0311e5243d69b73ed073e7ea441982e Mon Sep 17 00:00:00 2001
From: 游雁 <zhifu.gzf@alibaba-inc.com>
Date: 星期三, 27 三月 2024 16:05:29 +0800
Subject: [PATCH] train update
---
funasr/models/transducer/rnn_decoder.py | 18 +++++++++++-------
1 files changed, 11 insertions(+), 7 deletions(-)
diff --git a/funasr/models/transducer/rnn_decoder.py b/funasr/models/transducer/rnn_decoder.py
index 1743f99..b999d9c 100644
--- a/funasr/models/transducer/rnn_decoder.py
+++ b/funasr/models/transducer/rnn_decoder.py
@@ -1,14 +1,18 @@
-import random
+#!/usr/bin/env python3
+# -*- encoding: utf-8 -*-
+# Copyright FunASR (https://github.com/alibaba-damo-academy/FunASR). All Rights Reserved.
+# MIT License (https://opensource.org/licenses/MIT)
-import numpy as np
import torch
+import random
+import numpy as np
+import torch.nn as nn
import torch.nn.functional as F
+from funasr.register import tables
from funasr.models.transformer.utils.nets_utils import make_pad_mask
from funasr.models.transformer.utils.nets_utils import to_device
from funasr.models.language_model.rnn.attentions import initial_att
-from funasr.models.decoder.abs_decoder import AbsDecoder
-from funasr.utils.get_default_kwargs import get_default_kwargs
def build_attention_list(
@@ -79,8 +83,8 @@
)
return att_list
-
-class RNNDecoder(AbsDecoder):
+@tables.register("decoder_classes", "rnn_decoder")
+class RNNDecoder(nn.Module):
def __init__(
self,
vocab_size: int,
@@ -93,7 +97,7 @@
context_residual: bool = False,
replace_sos: bool = False,
num_encs: int = 1,
- att_conf: dict = get_default_kwargs(build_attention_list),
+ att_conf: dict = None,
):
# FIXME(kamo): The parts of num_spk should be refactored more more more
if rnn_type not in {"lstm", "gru"}:
--
Gitblit v1.9.1