fix a bug of encoder layer index

This commit is contained in:
Guo Liyong 2022-04-30 16:30:07 +08:00
parent 15220797e3
commit ac9655c450

View File

@ -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
)