mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-06 23:54:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

middle layer outputs from encoder

Browse Source
This commit is contained in:
Guo Liyong 2022-04-27 12:33:54 +08:00
parent 9ee57959ec
commit cbc9c50bfc
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/vq_pruned_transducer_stateless2/conformer.py
View File

@ -142,7 +142,7 @@ class Conformer(EncoderInterface):
if len(self.output_layers) > 1: if len(self.output_layers) > 1:
assert len(self.output_layers) == len(layers_result) assert len(self.output_layers) == len(layers_result)
return x, lengths, layers_result[0] return x, lengths, layers_result[0].permute(1, 0, 2)
return x, lengths return x, lengths