From 51f78f9b8d849ed4efbae9f5ac0de4fbe4043548 Mon Sep 17 00:00:00 2001 From: dohe0342 Date: Thu, 2 Feb 2023 14:00:54 +0900 Subject: [PATCH] from local --- .../ASR/conformer_ctc/.transformer.py.swp | Bin 69632 -> 69632 bytes egs/aishell/ASR/conformer_ctc/transformer.py | 2 +- 2 files changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/aishell/ASR/conformer_ctc/.transformer.py.swp b/egs/aishell/ASR/conformer_ctc/.transformer.py.swp index eac25f6945ac4d427e1082c00b76b104d19def54..e7728f426f04511e7430c402528f0b44a22101f0 100644 GIT binary patch delta 85 zcmZozz|ydQMKsAE%+puFQqO<^2m}}y(w%Q7U*0JC{yb|fHv_}d$($F0S(SN!tj&oR odblSazb!MF|BfkNNq$jshF(Z&UU7brj>6=|JIb5$?@TlX0GZYvqW}N^ delta 74 zcmZozz|ydQMKsAE%+puFQqO<^2m}}y+?;PGPuVE?{yb|lHv_}#$($F0Sv7fptj&oR ddblTx-8P;qbw_3L%Uj};C)^R)oPTGcF#un@8wCIW diff --git a/egs/aishell/ASR/conformer_ctc/transformer.py b/egs/aishell/ASR/conformer_ctc/transformer.py index 154fa848b..f35201b22 100644 --- a/egs/aishell/ASR/conformer_ctc/transformer.py +++ b/egs/aishell/ASR/conformer_ctc/transformer.py @@ -403,7 +403,7 @@ class TransfEncoder(nn.TransformerEncoder): def __init__(self, encoder_layer, num_layers, norm=None, enable_nested_tensor=True, mask_check=True): super(TransfEncoder, self).__init__() - def forward(self, src: Tensor, mask: Optional[torch.Tensor] = None, src_key_padding_mask: Optional[torch.Tensor] = None): + def forward(self, src: torch.Tensor, mask: Optional[torch.Tensor] = None, src_key_padding_mask: Optional[torch.Tensor] = None): """Pass the input through the encoder layers in turn. Args: