Remove debug statement

2022-10-14 20:59:26 +08:00 · 2022-10-14 20:59:26 +08:00 · 90953537ad
commit 90953537ad
parent 18ff1de337
1 changed files with 0 additions and 1 deletions
--- a/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py
@ -1239,7 +1239,6 @@ class RelPositionMultiheadAttention(nn.Module):
        with torch.no_grad():
            with torch.cuda.amp.autocast(enabled=False):
                attn_weights = attn_weights.to(torch.float32)
-                xyz
                attn_output = attn_output.to(torch.float32)
                attn_weights_entropy = -((attn_weights + 1.0e-20).log() * attn_weights).sum(
                    dim=-1).reshape(bsz, num_heads, seq_len).mean(dim=(0,2))