.

harvardnlp · Oct 14, 2021 · 2f72847 · 2f72847
1 parent ada7645
commit 2f72847
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/torch_struct/semimarkov.py b/torch_struct/semimarkov.py
@@ -61,7 +61,6 @@ def logpartition(self, log_potentials, lengths=None, force_grad=False):
         c[:, :, : K - 1, 0] = semiring.sum(
             torch.stack([c.data[:, :, : K - 1, 0], lp[:, :, 1:K]], dim=-1)
         )
-        end = torch.max(lengths) - 1
         mask = torch.zeros(*init.shape, device=log_potentials.device).bool()
         for k in range(1, K - 1):
             for b in range(batch):