From a9640da66f529382b584ea9c2dabf3ec420d7e34 Mon Sep 17 00:00:00 2001
From: leufen1 <l.leufen@fz-juelich.de>
Date: Fri, 12 Mar 2021 15:07:38 +0100
Subject: [PATCH] fix kernelsize for now

---
 mlair/model_modules/convolutional_networks.py | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/mlair/model_modules/convolutional_networks.py b/mlair/model_modules/convolutional_networks.py
index 2d8fd9e2..329d1952 100644
--- a/mlair/model_modules/convolutional_networks.py
+++ b/mlair/model_modules/convolutional_networks.py
@@ -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)
-- 
GitLab