mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-07 08:04:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

minor fix

Browse Source
This commit is contained in:
Dongji Gao 2023-09-21 12:23:06 -04:00
parent e964dc8960
commit 53623d713e
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
egs/librispeech/WSASR/conformer_ctc2/decode.py
View File

@ -160,8 +160,8 @@ def get_params() -> AttributeDict:
params = AttributeDict( params = AttributeDict(
{ {
# parameters for conformer # parameters for conformer
"subsampling_factor": 4, "subsampling_factor": 2,
"feature_dim": 80, "feature_dim": 768,
"nhead": 8, "nhead": 8,
"dim_feedforward": 2048, "dim_feedforward": 2048,
"encoder_dim": 512, "encoder_dim": 512,
@ -368,6 +368,7 @@ def decode_one_batch(
else: else:
assert False, f"Unsupported decoding method: {params.method}" assert False, f"Unsupported decoding method: {params.method}"
def decode_dataset( def decode_dataset(
dl: torch.utils.data.DataLoader, dl: torch.utils.data.DataLoader,
params: AttributeDict, params: AttributeDict,