From cc629c09a29515ca257bcb842efcfdc08a717ebc Mon Sep 17 00:00:00 2001 From: JinZr <60612200+JinZr@users.noreply.github.com> Date: Mon, 14 Aug 2023 20:14:50 +0800 Subject: [PATCH] updated --- .../zipformer_label_level_algn/alignment_attention_module.py | 3 ++- egs/librispeech/ASR/zipformer_label_level_algn/beam_search.py | 1 - 2 files changed, 2 insertions(+), 2 deletions(-) delete mode 120000 egs/librispeech/ASR/zipformer_label_level_algn/beam_search.py diff --git a/egs/librispeech/ASR/zipformer_label_level_algn/alignment_attention_module.py b/egs/librispeech/ASR/zipformer_label_level_algn/alignment_attention_module.py index 22a68aacf..86a2356b9 100644 --- a/egs/librispeech/ASR/zipformer_label_level_algn/alignment_attention_module.py +++ b/egs/librispeech/ASR/zipformer_label_level_algn/alignment_attention_module.py @@ -339,6 +339,7 @@ class RelPositionMultiheadAttentionWeights(nn.Module): use_pos_scores = True elif not self.training or random.random() >= float(self.pos_emb_skip_rate): use_pos_scores = True + use_pos_scores = False if use_pos_scores: pos_emb = self.linear_pos(pos_emb) @@ -385,7 +386,7 @@ class RelPositionMultiheadAttentionWeights(nn.Module): storage_offset=pos_scores.stride(3) * (lm_seq_len - 1), ) # print(pos_scores.shape) - attn_scores = attn_scores + pos_scores + # attn_scores = attn_scores + pos_scores if torch.jit.is_scripting() or torch.jit.is_tracing(): pass diff --git a/egs/librispeech/ASR/zipformer_label_level_algn/beam_search.py b/egs/librispeech/ASR/zipformer_label_level_algn/beam_search.py deleted file mode 120000 index 8554e44cc..000000000 --- a/egs/librispeech/ASR/zipformer_label_level_algn/beam_search.py +++ /dev/null @@ -1 +0,0 @@ -../pruned_transducer_stateless2/beam_search.py \ No newline at end of file