From 56f3657c1c07eaa2b617ebf5b2d7435d7f97faa7 Mon Sep 17 00:00:00 2001 From: leufen1 <l.leufen@fz-juelich.de> Date: Fri, 12 Mar 2021 15:24:49 +0100 Subject: [PATCH] changed dense layer --- mlair/model_modules/convolutional_networks.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/mlair/model_modules/convolutional_networks.py b/mlair/model_modules/convolutional_networks.py index 329d1952..c4a10990 100644 --- a/mlair/model_modules/convolutional_networks.py +++ b/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) -- GitLab