mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-08-11 02:52:18 +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:42:45 +08:00
parent f43ea5db30
commit 5cf4e2df07
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -592,7 +592,7 @@ def greedy_search(
1, context_size + 1
)
k[:, 0] = torch.sum(
k = torch.sum(
(
c[:, -context_size - 1 : -1]
== c[:, -1].expand_as(c[:, -context_size - 1 : -1])
@ -671,11 +671,9 @@ def greedy_search_batch(
# on which hyp[n][i] is decoded
timestamps = [[] for _ in range(N)]
decoder_input = torch.tensor(
hyps,
device=device,
dtype=torch.int64,
)[:, -context_size : ] # (N, context_size)
decoder_input = torch.tensor(hyps, device=device, dtype=torch.int64)[
:, -context_size:
] # (N, context_size)
k = torch.zeros(N, 1, device=device, dtype=torch.int64)
@ -720,10 +718,12 @@ def greedy_search_batch(
dtype=torch.int64,
)
k[:, 0] = torch.sum(
k = torch.sum(
(
c[:, : context_size]
== c[:, context_size : context_size + 1].expand_as(c[:, : context_size])
== c[:, context_size : context_size + 1].expand_as(
c[:, : context_size]
)
).int(),
dim=1,
keepdim=True,