Skip to content

Commit 23147de

Browse files
committed
Convs tests touch-ups
1 parent 7b72163 commit 23147de

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

tests/test_shape_inference.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -88,11 +88,11 @@ def test_Convolution1D():
8888

8989
def test_Convolution2D():
9090
for border_mode in ['same', 'valid']:
91-
for nb_row, nb_col in [(2, 2), (3, 3)]:
91+
for nb_row, nb_col in [(3, 3), (4, 4)]:
9292
for subsample in [(1, 1), (2, 2), (3, 3)]:
9393
if (subsample[0] > nb_row or subsample[1] > nb_col) and border_mode == 'same':
9494
continue
95-
for input_data_shape in [(2, 1, 4, 4), (2, 1, 5, 5)]:
95+
for input_data_shape in [(2, 1, 5, 5), (2, 1, 6, 6)]:
9696
layer = Convolution2D(nb_filter=1, nb_row=nb_row,
9797
nb_col=nb_row,
9898
border_mode=border_mode,
@@ -101,7 +101,7 @@ def test_Convolution2D():
101101
input_data = np.random.random(input_data_shape)
102102
check_layer_output_shape(layer, input_data)
103103

104-
for input_data_shape in [(2, 3, 3, 1)]:
104+
for input_data_shape in [(2, 5, 5, 1)]:
105105
layer = Convolution2D(nb_filter=1, nb_row=nb_row,
106106
nb_col=nb_row,
107107
border_mode=border_mode,

0 commit comments

Comments
 (0)