From 4e8c03bdfeab2b1211d2ebdc46f0a2a21823403a Mon Sep 17 00:00:00 2001 From: Qiyang Min Date: Tue, 12 Mar 2019 13:18:31 +0800 Subject: [PATCH] Imperative fix conv2d's attrs bugs (#16158) * Remove Desc in Forward Pass * Refactor VarBase * Add dbg info * Only check type in imperative mode * Polish code and support optimizer test=develop * Fix stop gradient problem in PyLayer test=develop * Fix conv2d's attrs test=develop --- python/paddle/fluid/imperative/nn.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/paddle/fluid/imperative/nn.py b/python/paddle/fluid/imperative/nn.py index 5aff3ea2d1f..bf3f7ce52e8 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, }) -- GitLab