From 847460bfe32b9d8ed0bc7f5f402c4c858a0a52bd Mon Sep 17 00:00:00 2001 From: zsdonghao Date: Wed, 9 May 2018 17:15:25 +0100 Subject: [PATCH] fix bug --- tensorlayer/layers/pooling.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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]