From 53623d713ef79b987b46e1a2ed10895372423a51 Mon Sep 17 00:00:00 2001 From: Dongji Gao Date: Thu, 21 Sep 2023 12:23:06 -0400 Subject: [PATCH] minor fix --- egs/librispeech/WSASR/conformer_ctc2/decode.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/egs/librispeech/WSASR/conformer_ctc2/decode.py b/egs/librispeech/WSASR/conformer_ctc2/decode.py index b883e448a..2d550a520 100755 --- a/egs/librispeech/WSASR/conformer_ctc2/decode.py +++ b/egs/librispeech/WSASR/conformer_ctc2/decode.py @@ -160,8 +160,8 @@ def get_params() -> AttributeDict: params = AttributeDict( { # parameters for conformer - "subsampling_factor": 4, - "feature_dim": 80, + "subsampling_factor": 2, + "feature_dim": 768, "nhead": 8, "dim_feedforward": 2048, "encoder_dim": 512, @@ -368,6 +368,7 @@ def decode_one_batch( else: assert False, f"Unsupported decoding method: {params.method}" + def decode_dataset( dl: torch.utils.data.DataLoader, params: AttributeDict,