official/mnist: Another formatting tweak

a763e45b · Asim Shankar · d1718519 · a763e45b
Commit a763e45b authored Apr 10, 2018 by Asim Shankar
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 2 deletions

official/mnist/mnist.py official/mnist/mnist.py +4 -2

No files found.
--- a/official/mnist/mnist.py
+++ b/official/mnist/mnist.py
@@ -66,13 +66,15 @@ def create_model(data_format):
              5,
              padding='same',
              data_format=data_format,
-              activation=tf.nn.relu), max_pool,
+              activation=tf.nn.relu),
+          max_pool,
          l.Conv2D(
              64,
              5,
              padding='same',
              data_format=data_format,
-              activation=tf.nn.relu), max_pool,
+              activation=tf.nn.relu),
+          max_pool,
          l.Flatten(),
          l.Dense(1024, activation=tf.nn.relu),
          l.Dropout(0.4),