Browse Source

Develop op MaxPoolWithArgMax

tags/v0.3.0-alpha
buxue chang zherui 6 years ago
parent
commit
824d9e4956
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/nn/layer/pooling.py

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

@@ -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):


Loading…
Cancel
Save