mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-12 11:32:19 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix for style check

Browse Source
This commit is contained in:
yifanyang 2023-02-06 18:51:43 +08:00
parent 3554ed32db
commit 06b583ee84
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
egs/librispeech/ASR/pruned_transducer_stateless9/beam_search.py
View File

@ -672,7 +672,7 @@ def greedy_search_batch(
timestamps = [[] for _ in range(N)]
decoder_input = torch.tensor(hyps, device=device, dtype=torch.int64)[
:, -context_size:
:, -context_size:
] # (N, context_size)
k = torch.zeros(N, 1, device=device, dtype=torch.int64)