diff --git a/tensorlayer/layers/convolution.py b/tensorlayer/layers/convolution.py index a103739fc..d7b3432d8 100644 --- a/tensorlayer/layers/convolution.py +++ b/tensorlayer/layers/convolution.py @@ -59,7 +59,7 @@ def __init__( act = tf.identity logging.info("Conv1dLayer %s: shape:%s stride:%s pad:%s act:%s" % (self.name, str(shape), str(stride), padding, act.__name__)) - with tf.variable_scope(name) as vs: + with tf.variable_scope(name): # as vs: W = tf.get_variable(name='W_conv1d', shape=shape, initializer=W_init, dtype=D_TYPE, **W_init_args) self.outputs = tf.nn.convolution( self.inputs, W, strides=(stride, ), padding=padding, dilation_rate=(dilation_rate, ), data_format=data_format) # 1.2 @@ -834,9 +834,9 @@ def _tf_batch_map_offsets(inputs, offsets, grid_offset): offset_params = [osparam for osparam in offset_layer.all_params if osparam not in layer.all_params] offset_layers = [oslayer for oslayer in offset_layer.all_layers if oslayer not in layer.all_layers] - self.all_params.extend(offset_params) - self.all_layers.extend(offset_layers) - self.all_drop.update(offset_layer.all_drop) + self.all_params.extend(list(offset_params)) + self.all_layers.extend(list(offset_layers)) + self.all_drop.update(dict(offset_layer.all_drop)) # this layer self.all_layers.extend([self.outputs])