mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-04 14:44:18 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix for black

Browse Source
This commit is contained in:
yifanyeung 2023-11-27 23:55:58 +08:00
parent 42e6efe4ee
commit 45daf1c73a
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
egs/aishell/ASR/zipformer_distinctK/decoder.py
View File

@ -125,10 +125,7 @@ class Decoder(nn.Module):
return embedding_out + (k / (1 + k)).unsqueeze(2) * self.repeat_param
def forward(
self,
y: torch.Tensor,
k: torch.Tensor,
need_pad: bool = True
self, y: torch.Tensor, k: torch.Tensor, need_pad: bool = True
) -> torch.Tensor:
"""
Args: