Update compressor.py

petuum · Nov 1, 2020 · 414b91b · 414b91b
1 parent 508dc32
commit 414b91b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/autodist/kernel/synchronization/compressor.py b/autodist/kernel/synchronization/compressor.py
@@ -300,5 +300,5 @@ def _modified_gram_schmidt(matrix):
 
             rest = matrix[:, (i + 1):]
             rest -= math_ops.reduce_sum_v1(v * rest, axis=0, keepdims=True) * v
-            matrix = array_ops.concat([matrix[:, :i], v, rest], axis=1)
+            matrix = array_ops.concat([matrix[:, :i], v, rest], 1)
         return matrix