From ac9655c4507c1afb72bc30c319398d07af8934a5 Mon Sep 17 00:00:00 2001 From: Guo Liyong Date: Sat, 30 Apr 2022 16:30:07 +0800 Subject: [PATCH] fix a bug of encoder layer index --- .../ASR/vq_pruned_transducer_stateless2/conformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/vq_pruned_transducer_stateless2/conformer.py b/egs/librispeech/ASR/vq_pruned_transducer_stateless2/conformer.py index 15c1e89aa..37037ea1c 100644 --- a/egs/librispeech/ASR/vq_pruned_transducer_stateless2/conformer.py +++ b/egs/librispeech/ASR/vq_pruned_transducer_stateless2/conformer.py @@ -95,7 +95,7 @@ class Conformer(EncoderInterface): extra_output_layer >= 0 and extra_output_layer < num_encoder_layers - 1 ) - self.output_layers.insert(extra_output_layer, 0) + self.output_layers.insert(0, extra_output_layer) self.encoder = ConformerEncoder( encoder_layer, num_encoder_layers, output_layers=self.output_layers )