.

harvardnlp · Nov 21, 2019 · 82706ce · 82706ce
1 parent 64d708d
commit 82706ce
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 2 deletions.
diff --git a/torch_struct/alignment.py b/torch_struct/alignment.py
@@ -304,7 +304,7 @@ def convert(chart, size):
                 chart[n-1] = convert(chart[n-1], size)
                 sparse = False
             size = int(size / 2)
-
+            print(n, sparse, chart.shape)
             chart[n] = merge(chart[n - 1], size, sparse=sparse)
 
             # charta[n] = merge(charta[n - 1], size,  sparse=False)

diff --git a/torch_struct/semirings/fast_semirings.py b/torch_struct/semirings/fast_semirings.py
@@ -18,7 +18,6 @@ def matmul_size(a, b):
 
 def broadcast(a, b):
     size = matmul_size(a, b)
-    print(size)
     a = a.expand(*size[:-2], a.shape[-2], a.shape[-1])
     b = b.expand(*size[:-2], b.shape[-2], b.shape[-1])
     a2 = a.contiguous().view(-1, a.shape[-2], a.shape[-1])