From 775fcd1b14d080f0b5c2e485a57f8ee68201e39b Mon Sep 17 00:00:00 2001
From: 北念 <lzr265946@alibaba-inc.com>
Date: 星期三, 29 三月 2023 17:14:29 +0800
Subject: [PATCH] fix ContextualBiasDecoder spell
---
funasr/models/decoder/contextual_decoder.py | 6 +++---
1 files changed, 3 insertions(+), 3 deletions(-)
diff --git a/funasr/models/decoder/contextual_decoder.py b/funasr/models/decoder/contextual_decoder.py
index 32f550a..3b462e7 100644
--- a/funasr/models/decoder/contextual_decoder.py
+++ b/funasr/models/decoder/contextual_decoder.py
@@ -74,7 +74,7 @@
return x, tgt_mask, x_self_attn, x_src_attn
-class ContexutalBiasDecoder(nn.Module):
+class ContextualBiasDecoder(nn.Module):
def __init__(
self,
size,
@@ -83,7 +83,7 @@
normalize_before=True,
):
"""Construct an DecoderLayer object."""
- super(ContexutalBiasDecoder, self).__init__()
+ super(ContextualBiasDecoder, self).__init__()
self.size = size
self.src_attn = src_attn
if src_attn is not None:
@@ -186,7 +186,7 @@
),
)
self.dropout = nn.Dropout(dropout_rate)
- self.bias_decoder = ContexutalBiasDecoder(
+ self.bias_decoder = ContextualBiasDecoder(
size=attention_dim,
src_attn=MultiHeadedAttentionCrossAtt(
attention_heads, attention_dim, src_attention_dropout_rate
--
Gitblit v1.9.1