changed dense layer

mlair/model_modules/convolutional_networks.py

@@ -82,7 +82,6 @@ class CNN(AbstractModelClass):
         Build the model.
         """
         x_input = keras.layers.Input(shape=self._input_shape)
-        kernel = (self.kernel_size, 1)
         x_in = keras.layers.Conv2D(filters=16, kernel_size=(73, 1),
                                    kernel_initializer=self.kernel_initializer,
                                    kernel_regularizer=self.kernel_regularizer)(x_input)
@@ -97,10 +96,10 @@ class CNN(AbstractModelClass):
                                    kernel_regularizer=self.kernel_regularizer)(x_in)
         x_in = self.activation()(x_in)
         x_in = keras.layers.Flatten()(x_in)
-        x_in = keras.layers.Dense(64, kernel_initializer=self.kernel_initializer,
+        x_in = keras.layers.Dense(128, kernel_initializer=self.kernel_initializer,
                                   kernel_regularizer=self.kernel_regularizer)(x_in)
         x_in = self.activation()(x_in)
-        x_in = keras.layers.Dense(16, kernel_initializer=self.kernel_initializer,
+        x_in = keras.layers.Dense(32, kernel_initializer=self.kernel_initializer,
                                   kernel_regularizer=self.kernel_regularizer)(x_in)
         x_in = self.activation()(x_in)
         x_in = keras.layers.Dense(self._output_shape)(x_in)