Fix for style check

This commit is contained in:
yifanyang 2023-02-06 18:47:36 +08:00
parent 5cf4e2df07
commit 3554ed32db

View File

@ -720,9 +720,9 @@ def greedy_search_batch(
k = torch.sum(
(
c[:, : context_size]
c[:, :context_size]
== c[:, context_size : context_size + 1].expand_as(
c[:, : context_size]
c[:, :context_size]
)
).int(),
dim=1,