diff --git a/egs/librispeech/ASR/pruned_transducer_stateless5/.conformer.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless5/.conformer.py.swp index 048f58734..6b909d661 100644 Binary files a/egs/librispeech/ASR/pruned_transducer_stateless5/.conformer.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless5/.conformer.py.swp differ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py index 8dd61b213..9b4b2a1ef 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless5/conformer.py @@ -197,8 +197,6 @@ class Conformer(EncoderInterface): x = x.permute(1, 0, 2) # (T, N, C) ->(N, T, C) - layer_output = [x.permute(1, 0, 2) for x in layer_output] - return x, lengths @torch.jit.export @@ -695,8 +693,6 @@ class ConformerEncoder(nn.Module): outputs = [] - layer_output = [] - for i, mod in enumerate(self.layers): output = mod( output, @@ -708,8 +704,6 @@ class ConformerEncoder(nn.Module): if i in self.aux_layers: outputs.append(output) - layer_output.append(output) - output = self.combiner(outputs) return output, layer_output