From 74d3911910cca624f38c9f51ca4bab6c97a548de Mon Sep 17 00:00:00 2001 From: Mayavan Date: Tue, 13 Jul 2021 20:15:33 -0400 Subject: [PATCH] Fix identity mapping kernel size The identity mapping kernel size should be 1. --- ML/TensorFlow/Basics/tutorial8_keras_subclassing.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)