dropout

556b2eee · sjjsmuel · 576a0596 · 556b2eee
Commit 556b2eee authored Apr 11, 2020 by sjjsmuel
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 2 deletions

classifier/Resnet152.py classifier/Resnet152.py +2 -2

No files found.
--- a/classifier/Resnet152.py
+++ b/classifier/Resnet152.py
@@ -31,9 +31,9 @@ class Resnet152(NetworkBase):

        x = AveragePooling2D((14, 14), name='avg_pool')(base_model.output)
        x = Flatten()(x)
-        x = Dropout(0.7)(x)
+        x = Dropout(0.5)(x)
        x = Dense(65536, activation='relu')(x)
-        x = Dropout(0.7)(x)
+        x = Dropout(0.5)(x)
        x = Dense(1024, activation='relu')(x)
        x = Dropout(0.7)(x)
        out = Dense(self.NUM_CLASSES, activation='softmax')(x)