Skip to content
Snippets Groups Projects
Commit 7e529068 authored by leufen1's avatar leufen1
Browse files

bigger kernel

parent 30c27e99
No related branches found
No related tags found
5 merge requests!319add all changes of dev into release v1.4.0 branch,!318Resolve "release v1.4.0",!300include cnn class,!271Resolve "create CNN model class",!259Draft: Resolve "WRF-Datahandler should inherit from SingleStationDatahandler"
Pipeline #62654 passed with warnings
......@@ -81,7 +81,7 @@ class CNN(AbstractModelClass):
Build the model.
"""
x_input = keras.layers.Input(shape=self._input_shape)
kernel = (1, 1)
kernel = (5, 1)
pad_size = PadUtils.get_padding_for_same(kernel)
x_in = Padding2D("SymPad2D")(padding=pad_size, name="SymPad1")(x_input)
x_in = keras.layers.Conv2D(filters=16, kernel_size=kernel,
......@@ -91,8 +91,8 @@ class CNN(AbstractModelClass):
x_in = keras.layers.Conv2D(filters=32, kernel_size=kernel,
kernel_initializer=self.kernel_initializer,
kernel_regularizer=self.kernel_regularizer)(x_in)
x_in = keras.layers.MaxPooling2D(kernel, strides=(1, 1), padding='valid')(x_in)
x_in = self.activation()(x_in)
x_in = Padding2D("SymPad2D")(padding=pad_size, name="SymPad2")(x_in)
x_in = keras.layers.Conv2D(filters=64, kernel_size=kernel,
kernel_initializer=self.kernel_initializer,
kernel_regularizer=self.kernel_regularizer)(x_in)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment