Skip to content

S
seminar-breakout
Commit 22375132 authored by see--'s avatar see-- Committed by Yuxin Wu
Browse files
Options

fix "mnist-keras.py" (#1207)

parent 80a68c59
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/keras/mnist-keras.py
View file @ 22375132
...@@ -42,7 +42,7 @@ def clear_tower0_name_scope(): ...@@ -42,7 +42,7 @@ def clear_tower0_name_scope():
def get_keras_model(): def get_keras_model():
with clear_tower0_name_scope(): with clear_tower0_name_scope():
M = keras.models.Sequential() M = keras.models.Sequential()
M.add(KL.Conv2D(32, 3, activation='relu', input_shape=[IMAGE_SIZE, IMAGE_SIZE, 1], padding='same')) M.add(KL.Conv2D(32, 3, activation='relu', padding='same'))
M.add(KL.BatchNormalization()) M.add(KL.BatchNormalization())
M.add(KL.MaxPooling2D()) M.add(KL.MaxPooling2D())
M.add(KL.Conv2D(32, 3, activation='relu', padding='same')) M.add(KL.Conv2D(32, 3, activation='relu', padding='same'))
......
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