Skip to content
Snippets Groups Projects

Resolve "release v1.4.0"

Merged Ghost User requested to merge release_v1.4.0 into master
1 file
+ 4
4
Compare changes
  • Side-by-side
  • Inline
@@ -83,16 +83,16 @@ class CNN(AbstractModelClass):
"""
x_input = keras.layers.Input(shape=self._input_shape)
kernel = (self.kernel_size, 1)
x_in = keras.layers.Conv2D(filters=16, kernel_size=kernel,
x_in = keras.layers.Conv2D(filters=16, kernel_size=(73, 1),
kernel_initializer=self.kernel_initializer,
kernel_regularizer=self.kernel_regularizer)(x_input)
x_in = self.activation()(x_in)
x_in = keras.layers.Conv2D(filters=32, kernel_size=kernel,
x_in = keras.layers.Conv2D(filters=32, kernel_size=(49, 1),
kernel_initializer=self.kernel_initializer,
kernel_regularizer=self.kernel_regularizer)(x_in)
x_in = self.activation()(x_in)
x_in = keras.layers.MaxPooling2D(kernel, strides=(1, 1), padding='valid')(x_in)
x_in = keras.layers.Conv2D(filters=64, kernel_size=kernel,
x_in = keras.layers.MaxPooling2D((25, 1), strides=(1, 1), padding='valid')(x_in)
x_in = keras.layers.Conv2D(filters=64, kernel_size=(13, 1),
kernel_initializer=self.kernel_initializer,
kernel_regularizer=self.kernel_regularizer)(x_in)
x_in = self.activation()(x_in)
Loading