Minor keyword change in train.py

swghosh · Sep 25, 2019 · 9b8a0d9 · 9b8a0d9
1 parent 97313a8
commit 9b8a0d9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/train.py b/train.py
@@ -80,7 +80,7 @@ def dense_layer(**args):
         lc2d_layer(filters=16, kernel_size=5, name='L6'),
         keras.layers.Flatten(name='F0'),
         dense_layer(units=4096, activation=keras.activations.relu, name='F7'),
-        keras.layers.Dropout(0.4, name='D0'),
+        keras.layers.Dropout(rate=0.4, name='D0'),
         dense_layer(units=NUM_CLASSES, activation=keras.activations.softmax, name='F8')
     ], name='DeepFace')
     deepface.summary()