From 3dbb8019af7fa0c4f59e0498ab53e66f901f6e85 Mon Sep 17 00:00:00 2001 From: Hunor Laczko Date: Fri, 30 Sep 2022 21:55:50 +0000 Subject: [PATCH] remove unnecesary part --- convnext/models/convnext_tf.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/convnext/models/convnext_tf.py b/convnext/models/convnext_tf.py index f413ea7..c5878be 100644 --- a/convnext/models/convnext_tf.py +++ b/convnext/models/convnext_tf.py @@ -140,6 +140,7 @@ class ConvNeXt(tf.keras.Model): def forward_features(self, x): for i in range(len(self.downsample_layers)): + # x = self.downsample_layers[i](x) x = self.downsample_layers[i][0](x) x = self.downsample_layers[i][1](x) x = self.stages[i](x) @@ -147,10 +148,6 @@ class ConvNeXt(tf.keras.Model): def call(self, x): x = self.forward_features(x) - if self.include_top: - x = self.avg(x) - x = self.norm(x) - x = self.head(x) return x