This is an automated email from the ASF dual-hosted git repository.
jxie 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 4c13b96 Update conv_layers.py (#7388)
4c13b96 is described below
commit 4c13b96b9ae73ac2809339798a923d2270c54f16
Author: Eric Junyuan Xie <piiswr...@users.noreply.github.com>
AuthorDate: Tue Aug 8 15:20:20 2017 -0700
Update conv_layers.py (#7388)
* Update conv_layers.py
* Update conv_layers.py
---
python/mxnet/gluon/nn/conv_layers.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/python/mxnet/gluon/nn/conv_layers.py
b/python/mxnet/gluon/nn/conv_layers.py
index d9608a1..9094edd 100644
--- a/python/mxnet/gluon/nn/conv_layers.py
+++ b/python/mxnet/gluon/nn/conv_layers.py
@@ -65,7 +65,7 @@ class _Conv(HybridBlock):
def __init__(self, channels, kernel_size, strides, padding, dilation,
groups, layout, in_channels=0, activation=None, use_bias=True,
weight_initializer=None, bias_initializer='zeros',
- op_name='Convolution', prefix=None, params=None, **kwargs):
+ op_name='Convolution', adj=None, prefix=None, params=None):
super(_Conv, self).__init__(prefix=prefix, params=params)
with self.name_scope():
self._channels = channels
@@ -81,7 +81,8 @@ class _Conv(HybridBlock):
'kernel': kernel_size, 'stride': strides, 'dilate': dilation,
'pad': padding, 'num_filter': channels, 'num_group': groups,
'no_bias': not use_bias, 'layout': layout}
- self._kwargs.update(kwargs)
+ if adj is not None:
+ self._kwargs['adj'] = adj
dshape = [0]*(len(kernel_size) + 2)
dshape[layout.find('N')] = 1
--
To stop receiving notification emails like this one, please contact
['"comm...@mxnet.apache.org" <comm...@mxnet.apache.org>'].
