fix karpathy/convnetjs#61

radioman · Nov 24, 2016 · 71edc02 · 71edc02
1 parent c0f3ed1
commit 71edc02
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ConvNetSharp/Layers/DropOutLayer.cs b/src/ConvNetSharp/Layers/DropOutLayer.cs
@@ -43,7 +43,7 @@ public override Volume Forward(Volume input, bool isTraining = false)
                 // scale the activations during prediction
                 for (var i = 0; i < length; i++)
                 {
-                    output.Weights[i] *= this.DropProb.Value;
+                    output.Weights[i] *= 1 - this.DropProb.Value;
                 }
             }