.

harvardnlp · Sep 9, 2019 · d65ddb5 · d65ddb5
1 parent 47fc6b2
commit d65ddb5
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torch_struct/deptree.py b/torch_struct/deptree.py
@@ -83,8 +83,8 @@ def sstack(a):
                 BC2 = torch.cat([AC_next[:, :, 1:].unsqueeze(-1), BC[:, :, 1:]], dim=3)
 
             start = semiring.dot(BC2[L], AC2[R])
-            arcs[k] = stack(semiring.times(start, arc_scores[:, f[1], f[0]]),
-                            semiring.times(start, arc_scores[:, f[0], f[1]]))
+            arcs[k] = stack(semiring.times(start),   #, arc_scores[:, f[1], f[0]]),
+                            semiring.times(start))  #, arc_scores[:, f[0], f[1]]))
 
             AIR2 = torch.cat([AIR[:, :-1], arcs[k][R].unsqueeze(-1)], dim=2)
             BIL2 = torch.cat([arcs[k][L].unsqueeze(-1), BIL[:, 1:]], dim=2)