remove unnecesary part

This commit is contained in:
2022-09-30 21:55:50 +00:00
parent ea5d80fa4a
commit 3dbb8019af
+1 -4
View File
@@ -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