smaller dense layer

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

Showing with 1 addition and 1 deletion

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

No files found.
--- a/classifier/Resnet152.py
+++ b/classifier/Resnet152.py
@@ -32,7 +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(65536, activation='relu')(x)
+        x = Dense(8192, activation='relu')(x)
        x = Dropout(0.5)(x)
        x = Dense(1024, activation='relu')(x)
        x = Dropout(0.7)(x)