Skip to content

Commit 5845fc0

Browse files
authored
fix bug (#557)
1 parent 8c848a1 commit 5845fc0

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tensorlayer/layers/pooling.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -201,7 +201,7 @@ def __init__(self, prev_layer, filter_size=(3, 3), strides=(2, 2), padding='SAME
201201
self.inputs, filter_size, strides, padding=padding, data_format='channels_last', name=name
202202
)
203203
else:
204-
if len(strides) == 2:
204+
if len(strides) != 2:
205205
raise Exception("len(strides) should be 2.")
206206
ksize = [1, filter_size[0], filter_size[1], 1]
207207
strides = [1, strides[0], strides[1], 1]
@@ -245,7 +245,7 @@ def __init__(self, prev_layer, filter_size=(3, 3), strides=(2, 2), padding='SAME
245245
self.inputs, filter_size, strides, padding=padding, data_format='channels_last', name=name
246246
)
247247
else:
248-
if len(strides) == 2:
248+
if len(strides) != 2:
249249
raise Exception("len(strides) should be 2.")
250250
ksize = [1, filter_size[0], filter_size[1], 1]
251251
strides = [1, strides[0], strides[1], 1]

0 commit comments

Comments
 (0)