Add assertion that filter sizes are equal

israelg99 · Feb 26, 2017 · e7b1324 · e7b1324
1 parent b9a28ad
commit e7b1324
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/eva/layers/masked_convolution2d.py b/eva/layers/masked_convolution2d.py
@@ -20,6 +20,7 @@ def build(self, input_shape):
 
         self.mask = np.ones(self.W_shape)
 
+        assert mask.shape[0] == mask.shape[1]
         filter_size = self.mask.shape[0]
         filter_center = filter_size / 2