Remove debug info

This commit is contained in:
Daniel Povey 2022-10-06 20:31:15 +08:00
parent a3179c30e7
commit bd325e8769

View File

@ -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