Update pooling.py (#31829)

Fix default argument of nn.MaxPool3D()
4 years ago · 5d89ec36dc
parent 649868ffb2
commit 5d89ec36dc
1 changed files with 2 additions and 2 deletions
--- a/python/paddle/nn/layer/pooling.py
+++ b/python/paddle/nn/layer/pooling.py
@ -589,8 +589,8 @@ class MaxPool3D(layers.Layer):

    def __init__(self,
                 kernel_size,
-                 stride,
-                 padding,
+                 stride=None,
+                 padding=0,
                 return_mask=False,
                 ceil_mode=False,
                 data_format="NCDHW",