Update deptree.py

harvardnlp · Mar 24, 2020 · a8ed72b · a8ed72b
1 parent 9f5a0f5
commit a8ed72b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torch_struct/deptree.py b/torch_struct/deptree.py
@@ -193,7 +193,7 @@ def deptree_part(arc_scores, eps=1e-5):
     lap = laplacian.masked_fill(eye != 0, 0)
     lap = -lap + torch.diag_embed(lap.sum(1), offset=0, dim1=-2, dim2=-1)
     lap[:, 0] = torch.diagonal(input, 0, -2, -1).exp()
-    return lap.det().log()
+    return lap.logdet()
 
 
 def deptree_nonproj(arc_scores, eps=1e-5):