Support single-root and multiple-root Eisner algorithm.

harvardnlp · Jul 27, 2020 · 35dbaa1 · 35dbaa1
1 parent c282a3d
commit 35dbaa1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_struct/deptree.py b/torch_struct/deptree.py
@@ -46,7 +46,7 @@ class DepTree(_Struct):
     """
 
     def _dp(self, arc_scores_in, lengths=None, force_grad=False, cache=True):
-        multiroot = getattr(self, "multiroot")
+        multiroot = getattr(self, "multiroot", True)
         if arc_scores_in.dim() not in (3, 4):
             raise ValueError("potentials must have dim of 3 (unlabeled) or 4 (labeled)")