Skip to content

GitLab

Commit a3efa3fe authored by Asim Shankar's avatar Asim Shankar
Browse files

[mnist]: Address another comment

parent 1fef2955
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
official/mnist/mnist.py
View file @ a3efa3fe
...@@ -100,7 +100,7 @@ class Model(object): ...@@ -100,7 +100,7 @@ class Model(object):
64, 5, padding='same', data_format=data_format, activation=tf.nn.relu) 64, 5, padding='same', data_format=data_format, activation=tf.nn.relu)
self.fc1 = tf.layers.Dense(1024, activation=tf.nn.relu) self.fc1 = tf.layers.Dense(1024, activation=tf.nn.relu)
self.fc2 = tf.layers.Dense(10) self.fc2 = tf.layers.Dense(10)
self.dropout = tf.layers.Dropout(0.5) self.dropout = tf.layers.Dropout(0.4)
self.max_pool2d = tf.layers.MaxPooling2D( self.max_pool2d = tf.layers.MaxPooling2D(
(2, 2), (2, 2), padding='same', data_format=data_format) (2, 2), (2, 2), padding='same', data_format=data_format)
......
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