This is an automated email from the ASF dual-hosted git repository.
zhasheng pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/master by this push:
new 3f10bd4 fix typo for assert messages in gluon/nn/conv_layers.py
(#9277)
3f10bd4 is described below
commit 3f10bd4e1114e34485b5114d07126e85ea43a320
Author: MinWoo Byeon <[email protected]>
AuthorDate: Tue Jan 2 16:08:33 2018 +0900
fix typo for assert messages in gluon/nn/conv_layers.py (#9277)
---
python/mxnet/gluon/nn/conv_layers.py | 9 +++++----
1 file changed, 5 insertions(+), 4 deletions(-)
diff --git a/python/mxnet/gluon/nn/conv_layers.py
b/python/mxnet/gluon/nn/conv_layers.py
index 645de98..822a81c 100644
--- a/python/mxnet/gluon/nn/conv_layers.py
+++ b/python/mxnet/gluon/nn/conv_layers.py
@@ -972,14 +972,15 @@ class GlobalMaxPool1D(_Pooling):
class GlobalMaxPool2D(_Pooling):
"""Global max pooling operation for spatial data."""
def __init__(self, layout='NCHW', **kwargs):
- assert layout == 'NCHW', "Only supports NCW layout for now"
+ assert layout == 'NCHW', "Only supports NCHW layout for now"
super(GlobalMaxPool2D, self).__init__(
(1, 1), None, 0, True, True, 'max', **kwargs)
+
class GlobalMaxPool3D(_Pooling):
"""Global max pooling operation for 3D data."""
def __init__(self, layout='NCDHW', **kwargs):
- assert layout == 'NCDHW', "Only supports NCW layout for now"
+ assert layout == 'NCDHW', "Only supports NCDHW layout for now"
super(GlobalMaxPool3D, self).__init__(
(1, 1, 1), None, 0, True, True, 'max', **kwargs)
@@ -995,7 +996,7 @@ class GlobalAvgPool1D(_Pooling):
class GlobalAvgPool2D(_Pooling):
"""Global average pooling operation for spatial data."""
def __init__(self, layout='NCHW', **kwargs):
- assert layout == 'NCHW', "Only supports NCW layout for now"
+ assert layout == 'NCHW', "Only supports NCHW layout for now"
super(GlobalAvgPool2D, self).__init__(
(1, 1), None, 0, True, True, 'avg', **kwargs)
@@ -1003,6 +1004,6 @@ class GlobalAvgPool2D(_Pooling):
class GlobalAvgPool3D(_Pooling):
"""Global max pooling operation for 3D data."""
def __init__(self, layout='NCDHW', **kwargs):
- assert layout == 'NCDHW', "Only supports NCW layout for now"
+ assert layout == 'NCDHW', "Only supports NCDHW layout for now"
super(GlobalAvgPool3D, self).__init__(
(1, 1, 1), None, 0, True, True, 'avg', **kwargs)
--
To stop receiving notification emails like this one, please contact
['"[email protected]" <[email protected]>'].
