From a02199df796769809728d0af524fe1552b6b453d Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Wed, 29 Mar 2023 20:33:43 +0800 Subject: [PATCH] Fix bug --- egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py b/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py index 37f4ecc3d..538646948 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py @@ -332,7 +332,7 @@ class Zipformer2(EncoderInterface): .reshape(num_frames_max * upsample_factor, batch_size, 3)) num_frames = (num_frames0 + ds - 1) // ds frame_mask = frame_mask[:num_frames] - channels = self.encoder_unmasked_dim[i] + channels = self.encoder_dim[i] feature_mask = torch.ones(num_frames, batch_size, channels, dtype=x.dtype, device=x.device) u1 = self.encoder_unmasked_dim[i]