diff --git a/mlair/model_modules/convolutional_networks.py b/mlair/model_modules/convolutional_networks.py index f9acdb72a2e4b1334d9b518a001155d3a5ce39c1..e7d1da23f7f9da3a80909ffdbc63ac7efe8a3ae3 100644 --- a/mlair/model_modules/convolutional_networks.py +++ b/mlair/model_modules/convolutional_networks.py @@ -83,7 +83,7 @@ class CNN(AbstractModelClass): x_input = keras.layers.Input(shape=self._input_shape) kernel = (1, 1) 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, kernel_initializer=self.kernel_initializer, kernel_regularizer=self.kernel_regularizer)(x_in) @@ -92,7 +92,7 @@ class CNN(AbstractModelClass): kernel_initializer=self.kernel_initializer, kernel_regularizer=self.kernel_regularizer)(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, kernel_initializer=self.kernel_initializer, kernel_regularizer=self.kernel_regularizer)(x_in)