From bd325e87696befb4cb9d2d8e759a22255d38bf85 Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Thu, 6 Oct 2022 20:31:15 +0800 Subject: [PATCH] Remove debug info --- egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py index 35386c9df..c4e1a597f 100644 --- a/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py +++ b/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py @@ -170,8 +170,6 @@ class Conformer(EncoderInterface): frame_mask0 = frame_mask1.unsqueeze(1).expand(num_frames1, ds, batch_size, 1).reshape( num_frames1 * ds, batch_size, 1)[:num_frames0] - print("frame_mask0 = ", frame_mask0.squeeze(-1)) - feature_mask0 = torch.ones(num_frames0, batch_size, self.d_model[0], dtype=x.dtype, device=x.device) feature_mask0[:, :, self.encoder_unmasked_dim:] *= frame_mask0