Skip to content
Snippets Groups Projects
Commit 30c27e99 authored by leufen1's avatar leufen1
Browse files

new pad layer names

parent 33940965
No related branches found
No related tags found
Loading
Pipeline #62653 passed
...@@ -83,7 +83,7 @@ class CNN(AbstractModelClass): ...@@ -83,7 +83,7 @@ class CNN(AbstractModelClass):
x_input = keras.layers.Input(shape=self._input_shape) x_input = keras.layers.Input(shape=self._input_shape)
kernel = (1, 1) kernel = (1, 1)
pad_size = PadUtils.get_padding_for_same(kernel) pad_size = PadUtils.get_padding_for_same(kernel)
x_in = Padding2D("SymPad2D")(padding=pad_size, name="SymPad")(x_input) x_in = Padding2D("SymPad2D")(padding=pad_size, name="SymPad1")(x_input)
x_in = keras.layers.Conv2D(filters=16, kernel_size=kernel, x_in = keras.layers.Conv2D(filters=16, kernel_size=kernel,
kernel_initializer=self.kernel_initializer, kernel_initializer=self.kernel_initializer,
kernel_regularizer=self.kernel_regularizer)(x_in) kernel_regularizer=self.kernel_regularizer)(x_in)
...@@ -92,7 +92,7 @@ class CNN(AbstractModelClass): ...@@ -92,7 +92,7 @@ class CNN(AbstractModelClass):
kernel_initializer=self.kernel_initializer, kernel_initializer=self.kernel_initializer,
kernel_regularizer=self.kernel_regularizer)(x_in) kernel_regularizer=self.kernel_regularizer)(x_in)
x_in = self.activation()(x_in) x_in = self.activation()(x_in)
x_in = Padding2D("SymPad2D")(padding=pad_size, name="SymPad")(x_in) x_in = Padding2D("SymPad2D")(padding=pad_size, name="SymPad2")(x_in)
x_in = keras.layers.Conv2D(filters=64, kernel_size=kernel, x_in = keras.layers.Conv2D(filters=64, kernel_size=kernel,
kernel_initializer=self.kernel_initializer, kernel_initializer=self.kernel_initializer,
kernel_regularizer=self.kernel_regularizer)(x_in) 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