diff --git a/tensorlayer/layers/pooling.py b/tensorlayer/layers/pooling.py index c3475a0ae..720e6809f 100644 --- a/tensorlayer/layers/pooling.py +++ b/tensorlayer/layers/pooling.py @@ -201,7 +201,7 @@ def __init__(self, prev_layer, filter_size=(3, 3), strides=(2, 2), padding='SAME self.inputs, filter_size, strides, padding=padding, data_format='channels_last', name=name ) else: - if len(strides) == 2: + if len(strides) != 2: raise Exception("len(strides) should be 2.") ksize = [1, filter_size[0], filter_size[1], 1] 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 self.inputs, filter_size, strides, padding=padding, data_format='channels_last', name=name ) else: - if len(strides) == 2: + if len(strides) != 2: raise Exception("len(strides) should be 2.") ksize = [1, filter_size[0], filter_size[1], 1] strides = [1, strides[0], strides[1], 1]