Develop op MaxPoolWithArgMax

824d9e49 · buxue · 高东海 · 6690a7fd · 824d9e49
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

mindspore/nn/layer/pooling.py mindspore/nn/layer/pooling.py +1 -1

未找到文件。
--- 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):