diff --git a/train.py b/train.py
index d59883d..8ed03cf 100644
--- a/train.py
+++ b/train.py
@@ -10,7 +10,7 @@ def create_model():
   return tf.keras.models.Sequential([
     tf.keras.layers.Flatten(input_shape=(28, 28)),
     tf.keras.layers.Dense(512, activation='relu'),
-    tf.keras.layers.Dropout(0.2),
+    tf.keras.layers.Dropout(0.5),
     tf.keras.layers.Dense(10, activation='softmax')
   ])