Added default value for dropout

parent 711641ba
......@@ -49,7 +49,7 @@ class LenetDropout(SequenceNetwork):
))
self.add(MaxPooling(name="pooling2"))
self.add(MaxPooling(name="pooling2"))
self.add(Dropout(name="dropout"))
self.add(Dropout(name="dropout", keep_prob=0.4))
self.add(FullyConnected(name="fc1", output_dim=fc1_output,
activation=tf.nn.tanh,
weights_initialization=Xavier(seed=seed, use_gpu=self.use_gpu),
......
Markdown is supported
0%
or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment