Commit 576a0596 authored by sjjsmuel's avatar sjjsmuel

bigger model and all trainable

parent a249842a
......@@ -29,16 +29,18 @@ class Resnet152(NetworkBase):
if shouldSave:
base_model.save('resnet_base_model.h5')
x = AveragePooling2D((7, 7), name='avg_pool')(base_model.output)
x = AveragePooling2D((14, 14), name='avg_pool')(base_model.output)
x = Flatten()(x)
x = Dropout(0.6)(x)
x = Dropout(0.7)(x)
x = Dense(65536, activation='relu')(x)
x = Dropout(0.7)(x)
x = Dense(1024, activation='relu')(x)
x = Dropout(0.6)(x)
x = Dropout(0.7)(x)
out = Dense(self.NUM_CLASSES, activation='softmax')(x)
model = Model(base_model.input, out)
for layer in model.layers[:-5]:
layer.trainable = False
#for layer in model.layers[:-5]:
# layer.trainable = False
return model
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