From 0468e1f21286371e31afed310a8ca4ee48b49732 Mon Sep 17 00:00:00 2001 From: reshalfahsi Date: Tue, 3 Jan 2023 20:35:36 +0700 Subject: [PATCH] Fix bug in residual_block --- model.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/model.py b/model.py index c11b32c..c28782a 100644 --- a/model.py +++ b/model.py @@ -23,7 +23,7 @@ def residual_block(x, num_filters): x = BatchNormalization()(x) s = Conv2D(num_filters, (1, 1), padding="same")(x_init) - s = BatchNormalization()(x) + s = BatchNormalization()(s) x = Add()([x, s]) x = Activation("relu")(x)