Commit 6be13c07 authored by sjjsmuel's avatar sjjsmuel

test new parameters

parent 0d44104b
......@@ -35,9 +35,10 @@ class Resnet50(NetworkBase):
layer.trainable = False
x = GlobalAveragePooling2D()(base_model.output)
x = Dropout(0.4)(x)
x = Dense(128, activation='relu', activity_regularizer=regularizers.l2(0.005))(x)
x = Dropout(0.2)(x)
x = Dense(64, activation='relu', activity_regularizer=regularizers.l2(0.01))(x)
out = Dense(self.NUM_CLASSES, activation='softmax', activity_regularizer=regularizers.l2(0.01), name='probs')(x)
out = Dense(self.NUM_CLASSES, activation='softmax', activity_regularizer=regularizers.l2(0.005), name='probs')(x)
model = Model(base_model.input, out)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment