From 8dd093e4ff840e8c3e17a9dc559d072fb4ccbf42 Mon Sep 17 00:00:00 2001 From: farizrahman4u Date: Sun, 19 Mar 2017 07:32:10 +0530 Subject: [PATCH] bug fix cos conv2d --- keras_contrib/layers/convolutional.py | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) diff --git a/keras_contrib/layers/convolutional.py b/keras_contrib/layers/convolutional.py index 613fef7..8674a3f 100644 --- a/keras_contrib/layers/convolutional.py +++ b/keras_contrib/layers/convolutional.py @@ -97,9 +97,8 @@ class Deconvolution3D(Convolution3D): # Arguments filters: Number of transposed convolution filters to use. - kernel_dim1: Length of the first dimension in the transposed convolution kernel. - kernel_dim2: Length of the second dimension in the transposed convolution kernel. - kernel_dim3: Length of the third dimension in the transposed convolution kernel. + kernel_size: kernel_size: An integer or tuple/list of 3 integers, specifying the + dimensions of the convolution window. output_shape: Output shape of the transposed convolution operation. tuple of integers `(nb_samples, filters, conv_dim1, conv_dim2, conv_dim3)`. @@ -258,8 +257,8 @@ class CosineConvolution2D(Layer): # Arguments filters: Number of convolution filters to use. - nb_row: Number of rows in the convolution kernel. - nb_col: Number of columns in the convolution kernel. + kernel_size: kernel_size: An integer or tuple/list of 2 integers, specifying the + dimensions of the convolution window. init: name of initialization function for the weights of the layer (see [initializers](../initializers.md)), or alternatively, Theano function to use for weights initialization.