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 error in how src was reshaped

Browse Source
This commit is contained in:
Daniel Povey 2023-05-16 17:19:47 +08:00
parent 0412d19f50
commit 5f5df4367d
1 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
egs/libriheavy/LM/zipformer1/subformer.py
View File

@ -684,8 +684,12 @@ class SubformerEncoder(nn.Module):
""" """
Reshape embeddings 'src' to have a different chunk size (in frames) Reshape embeddings 'src' to have a different chunk size (in frames)
""" """
num_channels = src.shape[-1] (seq_len, batch_size, num_channels) = src.shape
return src.reshape(chunk_size, -1, num_channels) num_chunks = src.shape[0] // chunk_size
src = src.reshape(num_chunks, chunk_size, batch_size, num_channels)
return src.contiguous().reshape(chunk_size, num_chunks * batch_size,
num_channels)
def _attn_offset_to_chunk_size(self, attn_offset: Tensor, batch_size: int, chunk_size: int) -> Tensor: def _attn_offset_to_chunk_size(self, attn_offset: Tensor, batch_size: int, chunk_size: int) -> Tensor:
""" """