Skip to content

Commit 2c22d37

Browse files
committed
Revert "Fix Head params to accept classifier_activation"
This reverts commit 00d4889.
1 parent 00d4889 commit 2c22d37

File tree

1 file changed

+3
-12
lines changed

1 file changed

+3
-12
lines changed

keras/applications/regnet.py

Lines changed: 3 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -833,12 +833,11 @@ def apply(inputs):
833833
return apply
834834

835835

836-
def Head(num_classes=1000, classifier_activation=None, name=None):
836+
def Head(num_classes=1000, name=None):
837837
"""Implementation of classification head of RegNet.
838838
839839
Args:
840840
num_classes: number of classes for Dense layer
841-
classifier_activation: activation function for the Dense layer
842841
name: name prefix
843842
844843
Returns:
@@ -849,11 +848,7 @@ def Head(num_classes=1000, classifier_activation=None, name=None):
849848

850849
def apply(x):
851850
x = layers.GlobalAveragePooling2D(name=name + "_head_gap")(x)
852-
x = layers.Dense(
853-
num_classes,
854-
activation=classifier_activation,
855-
name=name + "head_dense",
856-
)(x)
851+
x = layers.Dense(num_classes, name=name + "head_dense")(x)
857852
return x
858853

859854
return apply
@@ -982,12 +977,8 @@ def RegNet(
982977
in_channels = out_channels
983978

984979
if include_top:
980+
x = Head(num_classes=classes)(x)
985981
imagenet_utils.validate_activation(classifier_activation, weights)
986-
x = Head(
987-
num_classes=classes,
988-
classifier_activation=classifier_activation,
989-
name=model_name,
990-
)(x)
991982

992983
else:
993984
if pooling == "avg":

0 commit comments

Comments
 (0)