diff --git a/python/paddle/fluid/imperative/nn.py b/python/paddle/fluid/imperative/nn.py index 5aff3ea2d1f45a3ad0290a649bea06143bc1f146..bf3f7ce52e8d77fc2920b3bb63ec5b15a98b29b7 100644 --- a/python/paddle/fluid/imperative/nn.py +++ b/python/paddle/fluid/imperative/nn.py @@ -112,7 +112,7 @@ class Conv2D(layers.Layer): 'strides': self._stride, 'paddings': self._padding, 'dilations': self._dilation, - 'groups': self._groups, + 'groups': self._groups if self._groups else 1, 'use_cudnn': self._use_cudnn, 'use_mkldnn': False, })