diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.conformer_randomcombine.py.swp b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.conformer_randomcombine.py.swp index 8f462e5cf..f3bef4fdf 100644 Binary files a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.conformer_randomcombine.py.swp and b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/.conformer_randomcombine.py.swp differ diff --git a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer_randomcombine.py b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer_randomcombine.py index 83846158a..069c8c9a4 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer_randomcombine.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless_gtrans/conformer_randomcombine.py @@ -195,7 +195,7 @@ class Conformer(EncoderInterface): num_left_chunks=self.num_left_chunks, device=x.device, ) - x, layer_output = self.encoder( + x, layer_outputs = self.encoder( x, pos_emb, mask=mask, @@ -203,7 +203,7 @@ class Conformer(EncoderInterface): warmup=warmup, ) # (T, N, C) else: - x, layer_output = self.encoder( + x, layer_outputs = self.encoder( x, pos_emb, src_key_padding_mask=src_key_padding_mask,