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( k = torch.sum(
( (
c[:, : context_size] c[:, :context_size]
== c[:, context_size : context_size + 1].expand_as( == c[:, context_size : context_size + 1].expand_as(
c[:, : context_size] c[:, :context_size]
) )
).int(), ).int(),
dim=1, dim=1,