diff --git a/ML/TensorFlow/Basics/tutorial8_keras_subclassing.py b/ML/TensorFlow/Basics/tutorial8_keras_subclassing.py index 714b4ba6..d76b1874 100644 --- a/ML/TensorFlow/Basics/tutorial8_keras_subclassing.py +++ b/ML/TensorFlow/Basics/tutorial8_keras_subclassing.py @@ -43,7 +43,7 @@ def __init__(self, channels): self.cnn2 = CNNBlock(channels[1], 3) self.cnn3 = CNNBlock(channels[2], 3) self.pooling = layers.MaxPooling2D() - self.identity_mapping = layers.Conv2D(channels[1], 3, padding="same") + self.identity_mapping = layers.Conv2D(channels[1], 1, padding="same") def call(self, input_tensor, training=False): x = self.cnn1(input_tensor, training=training)