We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 7b72163 commit 23147deCopy full SHA for 23147de
tests/test_shape_inference.py
@@ -88,11 +88,11 @@ def test_Convolution1D():
88
89
def test_Convolution2D():
90
for border_mode in ['same', 'valid']:
91
- for nb_row, nb_col in [(2, 2), (3, 3)]:
+ for nb_row, nb_col in [(3, 3), (4, 4)]:
92
for subsample in [(1, 1), (2, 2), (3, 3)]:
93
if (subsample[0] > nb_row or subsample[1] > nb_col) and border_mode == 'same':
94
continue
95
- for input_data_shape in [(2, 1, 4, 4), (2, 1, 5, 5)]:
+ for input_data_shape in [(2, 1, 5, 5), (2, 1, 6, 6)]:
96
layer = Convolution2D(nb_filter=1, nb_row=nb_row,
97
nb_col=nb_row,
98
border_mode=border_mode,
@@ -101,7 +101,7 @@ def test_Convolution2D():
101
input_data = np.random.random(input_data_shape)
102
check_layer_output_shape(layer, input_data)
103
104
- for input_data_shape in [(2, 3, 3, 1)]:
+ for input_data_shape in [(2, 5, 5, 1)]:
105
106
107
0 commit comments