tonyyang-svail-patch-1
chengduoZH 7 years ago
parent d4dabe3e0b
commit 93e0609f1d

@ -1262,7 +1262,7 @@ def conv2d(input,
raise ValueError("use_cudnn should be True or False")
input_shape = input.shape
filter_shape = [num_filters, num_filter_channels] + filter_size
filter_shape = [num_filters, num_filter_channels] + list(filter_size)
def _get_default_param_initializer():
std = (2.0 / (filter_size[0]**2 * num_channels))**0.5
@ -1826,7 +1826,7 @@ def conv2d_transpose(input,
elif isinstance(filter_size, int):
filter_size = [filter_size, filter_size]
filter_shape = [input_channel, num_filters] + filter_size
filter_shape = [input_channel, num_filters] + list(filter_size)
img_filter = helper.create_parameter(
dtype=input.dtype, shape=filter_shape, attr=helper.param_attr)

Loading…
Cancel
Save