Added default value for dropout

parent 711641ba
...@@ -49,7 +49,7 @@ class LenetDropout(SequenceNetwork): ...@@ -49,7 +49,7 @@ class LenetDropout(SequenceNetwork):
)) ))
self.add(MaxPooling(name="pooling2")) self.add(MaxPooling(name="pooling2"))
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, self.add(FullyConnected(name="fc1", output_dim=fc1_output,
activation=tf.nn.tanh, activation=tf.nn.tanh,
weights_initialization=Xavier(seed=seed, use_gpu=self.use_gpu), 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