Develop op MaxPoolWithArgMax

5 years ago · 824d9e4956
parent 6690a7fd7a
commit 824d9e4956
1 changed files with 1 additions and 1 deletions
--- a/mindspore/nn/layer/pooling.py
+++ b/mindspore/nn/layer/pooling.py
@ -58,7 +58,7 @@ class _PoolNd(Cell):
        pass

    def extend_repr(self):
-        return 'kernel_size={kernel_size}, stride={stride}, pad_mode={pad_mode}'.format(**self.__dict__)
+        return 'kernel_size={kernel_size}, strides={strides}, pad_mode={pad_mode}'.format(**self.__dict__)


 class MaxPool2d(_PoolNd):