Skip to content

GitLab

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

official/mnist: Address comment in PR

parent a763e45b
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
official/mnist/mnist.py
View file @ cfe945ef
...@@ -58,6 +58,8 @@ def create_model(data_format): ...@@ -58,6 +58,8 @@ def create_model(data_format):
l = tf.keras.layers l = tf.keras.layers
max_pool = l.MaxPooling2D( max_pool = l.MaxPooling2D(
(2, 2), (2, 2), padding='same', data_format=data_format) (2, 2), (2, 2), padding='same', data_format=data_format)
# The model consists of a sequential chain of layers, so tf.keras.Sequential
# (a subclass of tf.keras.Model) makes for a compact description.
return tf.keras.Sequential( return tf.keras.Sequential(
[ [
l.Reshape(input_shape), l.Reshape(input_shape),
......
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