From 227da6e7205e0470ebf377d1b7403e8c94ea1f7a Mon Sep 17 00:00:00 2001 From: zhaojichen Date: Tue, 21 Apr 2020 10:27:26 -0400 Subject: [PATCH] add AvgPooling layer --- mindspore/nn/layer/pooling.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/mindspore/nn/layer/pooling.py b/mindspore/nn/layer/pooling.py index a19ef06b7e..28826c88bb 100644 --- a/mindspore/nn/layer/pooling.py +++ b/mindspore/nn/layer/pooling.py @@ -268,13 +268,8 @@ class AvgPool1d(_PoolNd): ParamValidator.check_type('kernel_size', kernel_size, [int,]) ParamValidator.check_type('stride', stride, [int,]) self.pad_mode = ParamValidator.check_string('pad_mode', pad_mode.upper(), ['VALID', 'SAME']) - if not isinstance(kernel_size, int): - ParamValidator.check_integer("kernel_size", kernel_size, 1, Rel.GE) - raise ValueError("kernel_size should be 1 int number but got {}". - format(kernel_size)) - if not isinstance(stride, int): - ParamValidator.check_integer("stride", stride, 1, Rel.GE) - raise ValueError("stride should be 1 int number but got {}".format(stride)) + ParamValidator.check_integer("kernel_size", kernel_size, 1, Rel.GE) + ParamValidator.check_integer("stride", stride, 1, Rel.GE) self.kernel_size = (1, kernel_size) self.stride = (1, stride) self.avg_pool = P.AvgPool(ksize=self.kernel_size,