mirrors/icefall
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

minor fix doc

Browse Source
This commit is contained in:
yaozengwei 2022-06-11 22:44:06 +08:00
parent 64c9053e39
commit 7f94b86bb0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/librispeech/ASR/conv_emformer_transducer_stateless/emformer.py
View File

@ -122,7 +122,7 @@ def stack_states(
conv_caches = [] conv_caches = []
for layer in state_list[0][1]: for layer in state_list[0][1]:
if batch_size > 1: if batch_size > 1:
# Note: We will stack conv_caches[layer][] later to get attn_caches[layer] # noqa # Note: We will stack conv_caches[layer][] later to get conv_caches[layer] # noqa
conv_caches.append([layer]) conv_caches.append([layer])
else: else:
conv_caches.append(layer.unsqueeze(0)) conv_caches.append(layer.unsqueeze(0))