luni/cvpr
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

remove unnecesary part

Browse Source
This commit is contained in:
luni
2022-09-30 21:55:50 +00:00
parent ea5d80fa4a
commit 3dbb8019af
1 changed files with 1 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
convnext/models/convnext_tf.py
+1 -4
View File
@@ -140,6 +140,7 @@ class ConvNeXt(tf.keras.Model):
def forward_features(self, x): def forward_features(self, x):
for i in range(len(self.downsample_layers)): 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][0](x)
x = self.downsample_layers[i][1](x) x = self.downsample_layers[i][1](x)
x = self.stages[i](x) x = self.stages[i](x)
@@ -147,10 +148,6 @@ class ConvNeXt(tf.keras.Model):
def call(self, x): def call(self, x):
x = self.forward_features(x) x = self.forward_features(x)
if self.include_top:
x = self.avg(x)
x = self.norm(x)
x = self.head(x)
return x return x