Commit a52bb4ed authored by sjjsmuel's avatar sjjsmuel

dropout missed

parent 3a9ea9c0
......@@ -35,7 +35,7 @@ class Resnet152(NetworkBase):
x = Dense(8192, activation='relu')(x)
x = Dropout(0.5)(x)
x = Dense(1024, activation='relu')(x)
x = Dropout(0.7)(x)
x = Dropout(0.5)(x)
out = Dense(self.NUM_CLASSES, activation='softmax')(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