.

harvardnlp · Nov 26, 2019 · 6cae984 · 6cae984
1 parent 5ee12a9
commit 6cae984
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_struct/semimarkov.py b/torch_struct/semimarkov.py
@@ -77,7 +77,7 @@ def _dp(self, log_potentials, lengths=None, force_grad=False):
             chart = semiring.matmul(chart[:, :, 1::2], chart[:, :, 0::2])
 
         final = chart.view(-1, batch, 1, K_1, C, K_1, C)
-        # v = semiring.sum(semiring.sum(final[:, :, 0, 0, :, 0, :]))
+        v = semiring.sum(semiring.sum(final[:, :, 0, 0, :, 0, :]))
         return v, [log_potentials], None
 
     # def _dp_standard(self, edge, lengths=None, force_grad=False):