From 7f94b86bb032c60b4bc948bb084faa2433cb8e44 Mon Sep 17 00:00:00 2001 From: yaozengwei Date: Sat, 11 Jun 2022 22:44:06 +0800 Subject: [PATCH] minor fix doc --- .../ASR/conv_emformer_transducer_stateless/emformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/conv_emformer_transducer_stateless/emformer.py b/egs/librispeech/ASR/conv_emformer_transducer_stateless/emformer.py index 39fc04186..d8c453ca6 100644 --- a/egs/librispeech/ASR/conv_emformer_transducer_stateless/emformer.py +++ b/egs/librispeech/ASR/conv_emformer_transducer_stateless/emformer.py @@ -122,7 +122,7 @@ def stack_states( conv_caches = [] for layer in state_list[0][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]) else: conv_caches.append(layer.unsqueeze(0))