From 4dae57e381de56f823625f4b48269de6cdbe8f28 Mon Sep 17 00:00:00 2001 From: leufen1 <l.leufen@fz-juelich.de> Date: Mon, 29 Mar 2021 14:31:08 +0200 Subject: [PATCH] use dropout in CNNs --- mlair/model_modules/convolutional_networks.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/mlair/model_modules/convolutional_networks.py b/mlair/model_modules/convolutional_networks.py index d4955d3d..624cfa09 100644 --- a/mlair/model_modules/convolutional_networks.py +++ b/mlair/model_modules/convolutional_networks.py @@ -104,11 +104,15 @@ class CNN(AbstractModelClass): kernel_initializer=self.kernel_initializer, kernel_regularizer=self.kernel_regularizer)(x_in) x_in = self.activation()(x_in) + if self.dropout is not None: + x_in = self.dropout(self.dropout_rate)(x_in) 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) + if self.dropout is not None: + x_in = self.dropout(self.dropout_rate)(x_in) x_in = keras.layers.Flatten()(x_in) x_in = keras.layers.Dense(128, kernel_initializer=self.kernel_initializer, kernel_regularizer=self.kernel_regularizer)(x_in) -- GitLab