提交 4e8c03bd 编写于 作者: Q Qiyang Min 提交者: Jiabin Yang

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
上级 08e75731
......@@ -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,
})
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册