aymericdamien 7 anni fa
parent
commit
971c96b7b2

+ 1 - 1
examples/3_NeuralNetworks/convolutional_network.py

@@ -26,7 +26,7 @@ batch_size = 128
 # Network Parameters
 num_input = 784 # MNIST data input (img shape: 28*28)
 num_classes = 10 # MNIST total classes (0-9 digits)
-dropout = 0.75 # Dropout, probability to keep units
+dropout = 0.25 # Dropout, probability to drop a unit
 
 
 # Create the neural network

+ 1 - 1
notebooks/3_NeuralNetworks/convolutional_network.ipynb

@@ -74,7 +74,7 @@
     "# Network Parameters\n",
     "num_input = 784 # MNIST data input (img shape: 28*28)\n",
     "num_classes = 10 # MNIST total classes (0-9 digits)\n",
-    "dropout = 0.75 # Dropout, probability to keep units"
+    "dropout = 0.25 # Dropout, probability to drop a unit"
    ]
   },
   {