Commit a249842a authored by sjjsmuel's avatar sjjsmuel

try to optimise training parameters

parent d8a97133
......@@ -31,8 +31,9 @@ class Resnet152(NetworkBase):
x = AveragePooling2D((7, 7), name='avg_pool')(base_model.output)
x = Flatten()(x)
x = Dropout(0.5)(x)
x = Dropout(0.6)(x)
x = Dense(1024, activation='relu')(x)
x = Dropout(0.6)(x)
out = Dense(self.NUM_CLASSES, activation='softmax')(x)
model = Model(base_model.input, out)
......
......@@ -92,7 +92,7 @@ model = network.get_model()
#compile the model
#model.compile(optimizer='rmsprop', loss='categorical_crossentropy', metrics=['accuracy'])
model.compile(optimizer=Adam(lr=0.0001), loss='categorical_crossentropy', metrics=['accuracy'])
model.compile(optimizer=Adam(lr=0.000001), loss='categorical_crossentropy', metrics=['accuracy'])
#Adam(lr=0.0001)
# Was ist mir Dropout einfügen?!
'''
......
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