Commit 4741732d authored by sjjsmuel's avatar sjjsmuel

smaller but more trainiable

parent 8fec60a5
......@@ -29,10 +29,10 @@ class Resnet152(NetworkBase):
if shouldSave:
base_model.save('resnet_base_model.h5')
x = AveragePooling2D((14, 14), name='avg_pool')(base_model.output)
x = AveragePooling2D((7, 7), name='avg_pool')(base_model.output)
x = Flatten()(x)
x = Dropout(0.5)(x)
x = Dense(1024, activation='relu')(x)
x = Dense(2048, activation='relu')(x)
x = Dropout(0.5)(x)
out = Dense(self.NUM_CLASSES, activation='softmax')(x)
......
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