'fix_dims_mismatching'

vislearn · Jan 28, 2021 · 550257b · 550257b
1 parent 4a5c590
commit 550257b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/FrEIA/modules/coupling_layers.py b/FrEIA/modules/coupling_layers.py
@@ -20,7 +20,8 @@ def __init__(self, dims_in, dims_c=[], subnet_constructor=None):
         self.split_len1 = channels // 2
         self.split_len2 = channels - channels // 2
 
-        assert all([dims_c[i][1:] == dims_in[0][1:] for i in range(len(dims_c))]), \
+        # assert all([dims_c[i][1:] == dims_in[0][1:] for i in range(len(dims_c))]), \
+        assert all([tuple(dims_c[i][1:]) == tuple(dims_in[0][1:]) for i in range(len(dims_c))]), \
             "Dimensions of input and one or more conditions don't agree."
         self.conditional = (len(dims_c) > 0)
         condition_length = sum([dims_c[i][0] for i in range(len(dims_c))])