reduce again.

f586f6e3 · sjjsmuel · a52bb4ed · f586f6e3
Commit f586f6e3 authored Apr 11, 2020 by sjjsmuel
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 3 deletions

classifier/Resnet152.py classifier/Resnet152.py +1 -3

No files found.
--- a/classifier/Resnet152.py
+++ b/classifier/Resnet152.py
@@ -32,9 +32,7 @@ class Resnet152(NetworkBase):
        x = AveragePooling2D((14, 14), name='avg_pool')(base_model.output)
        x = Flatten()(x)
        x = Dropout(0.5)(x)
-        x = Dense(8192, activation='relu')(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)