.

harvardnlp · Nov 26, 2019 · b853722 · b853722
1 parent 3486167
commit b853722
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torch_struct/semimarkov.py b/torch_struct/semimarkov.py
@@ -61,8 +61,8 @@ def _dp(self, log_potentials, lengths=None, force_grad=False):
                          lp[:, :, 1:K]], dim=-1)
         )
         end = torch.min(lengths) - 1
-        for k in range(1, K - 1):
-            semiring.one_(init.data[:, :, : end - (k - 1), k - 1, k].diagonal(0, -2, -1))
+        # for k in range(1, K - 1):
+        #     semiring.one_(init.data[:, :, : end - (k - 1), k - 1, k].diagonal(0, -2, -1))
 
         K_1 = K - 1