From 90953537adb2e5427c799d3a859026368f118b8d Mon Sep 17 00:00:00 2001 From: Daniel Povey Date: Fri, 14 Oct 2022 20:59:26 +0800 Subject: [PATCH] Remove debug statement --- egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py | 1 - 1 file changed, 1 deletion(-) diff --git a/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py b/egs/librispeech/ASR/pruned_transducer_stateless7/conformer.py index b04a695cd..798c316a5 100644 --- 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))