From 6cae984842c141073d2c8f494beb053eb4f6cfe3 Mon Sep 17 00:00:00 2001 From: Sasha Date: Tue, 26 Nov 2019 12:04:27 -0500 Subject: [PATCH] . --- torch_struct/semimarkov.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/torch_struct/semimarkov.py b/torch_struct/semimarkov.py index 4fcdb54d..db556c7a 100644 --- 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):