wassname/keras-contrib
1
0
Fork 0
mirror of https://github.com/wassname/keras-contrib.git synced 2026-08-16 11:22:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

filter shape bug fix

Browse Source
This commit is contained in:
Michael Oliver
2017-02-22 18:08:22 -08:00
parent 36c0c9e600
commit 5b273df036
1 changed files with 5 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
keras_contrib/layers/convolutional.py
+5 -5
View File
@@ -395,23 +395,23 @@ class CosineConvolution2D(Layer):
return (input_shape[0], rows, cols, self.nb_filter)
def call(self, x, mask=None):
b, xb = 0, 0
b, xb = 0., 0.
if self.dim_ordering == 'th':
W_sum_axes = [1, 2, 3]
if self.bias:
b = K.reshape(self.b, (self.nb_filter, 1, 1, 1))
xb = 1
xb = 1.
elif self.dim_ordering == 'tf':
W_sum_axes = [0, 1, 2]
if self.bias:
b = K.reshape(self.b, (1, 1, 1, self.nb_filter))
xb = 1
xb = 1.
Wnorm = K.sqrt(K.sum(K.square(self.W), axis=W_sum_axes, keepdims=True) + K.square(b) + K.epsilon())
xnorm = K.sqrt(K.conv2d(x**2, self.W_norm, strides=self.subsample,
xnorm = K.sqrt(K.conv2d(K.square(x), self.W_norm, strides=self.subsample,
border_mode=self.border_mode,
dim_ordering=self.dim_ordering,
filter_shape=self.W_shape) + xb + K.epsilon())
filter_shape=self.W_norm_shape) + xb + K.epsilon())
W = self.W / Wnorm