mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-19 05:54:20 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix bug, >> -> <<

Browse Source
This commit is contained in:
yaozengwei 2022-07-08 16:45:12 +08:00
parent 5cfdbd3699
commit 2057124004
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/librispeech/ASR/conv_emformer_transducer_stateless/emformer.py
View File

@ -1585,7 +1585,7 @@ class EmformerEncoder(nn.Module):
memory_mask = (
(
(
num_processed_frames << int(math.log(self.chunk_length, 2))
num_processed_frames >> int(math.log(self.chunk_length, 2))
).view(x.size(1), 1)
<= torch.arange(self.memory_size, device=x.device).expand(
x.size(1), self.memory_size

2
egs/librispeech/ASR/conv_emformer_transducer_stateless2/emformer.py
View File

@ -1493,7 +1493,7 @@ class EmformerEncoder(nn.Module):
memory_mask = (
(
(
num_processed_frames << int(math.log(self.chunk_length, 2))
num_processed_frames >> int(math.log(self.chunk_length, 2))
).view(x.size(1), 1)
<= torch.arange(self.memory_size, device=x.device).expand(
x.size(1), self.memory_size