Commit 982351b1 authored by sjjsmuel's avatar sjjsmuel

dropout

parent cfbc26cd
......@@ -29,7 +29,8 @@ class Resnet50(NetworkBase):
if shouldSave:
base_model.save('data/resnet_50_base_model.h5')
x = GlobalAveragePooling2D(base_model.output)
x = GlobalAveragePooling2D()(base_model.output)
x = Dropout(0.5)(x)
out = Dense(self.NUM_CLASSES, activation='softmax', 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