Revert making the dropout of pos_emb independent across the batch.

2025-12-11 06:55:27 +00:00 · 2022-11-14 15:34:39 +08:00 · 2022-11-14 15:34:39 +08:00 · ce4b50d094
commit ce4b50d094
parent 804917837e
1 changed files with 1 additions and 2 deletions
--- a/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py
+++ b/egs/librispeech/ASR/pruned_transducer_stateless7/zipformer.py
@ -984,8 +984,7 @@ class RelPositionalEncoding(torch.nn.Module):
            + x.size(0),
            :
        ]
-        batch_size = x.size(1)
-        pos_emb = pos_emb.unsqueeze(0).expand(batch_size, -1, -1)  # now: (batch_size, 2*time-1, embed_dim)
+        pos_emb = pos_emb.unsqueeze(0)
        return self.dropout(pos_emb)