From 8e650a584134e9cd42216427ac1f2f2a0ae45b74 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Mon, 27 Sep 2021 12:19:01 +0800 Subject: [PATCH] Update egs/librispeech/ASR/conformer_lm/conformer.py Co-authored-by: Fangjun Kuang --- egs/librispeech/ASR/conformer_lm/conformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/conformer_lm/conformer.py b/egs/librispeech/ASR/conformer_lm/conformer.py index fe0a5eec9..d08b64ea3 100644 --- a/egs/librispeech/ASR/conformer_lm/conformer.py +++ b/egs/librispeech/ASR/conformer_lm/conformer.py @@ -112,7 +112,7 @@ class MaskedLmConformer(nn.Module): Returns: Returns (memory, pos_emb), where: - `memory` is a Tensor containing the encoded data; it is of shape (N, T, C) + `memory` is a Tensor containing the encoded data; it is of shape (T, N, C) where C is the embedding_dim. `pos_emb` is a Tensor containing the relative positional encoding, of shape (1, 2*T-1, C)