Retain the computational graph for marginals

yzhangcs · yzhangcs · commit 4f8a0d26713b · 2021-12-27T19:06:57.000+08:00
diff --git a/supar/structs/dist.py b/supar/structs/dist.py
@@ -127,4 +127,4 @@ def forward(self, semiring):
     def backward(self, log_partition):
         return autograd.grad(log_partition,
                              self.scores if isinstance(self.scores, torch.Tensor) else self.scores[0],
-                             retain_graph=True)[0]
+                             create_graph=True)[0]
