Update conv_layers.py (#7388)

* Update conv_layers.py * Update conv_layers.py
apache · Aug 8, 2017 · 4c13b96 · 4c13b96
1 parent 07661ae
commit 4c13b96
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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 @@ def __init__(self, channels, kernel_size, strides, padding, dilation,
                 '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