Skip to content

Instantly share code, notes, and snippets.

@rafalpronko
Created February 11, 2019 05:51
Show Gist options
  • Save rafalpronko/392fea24b2ca4c3bc835aab2d20d3920 to your computer and use it in GitHub Desktop.
Save rafalpronko/392fea24b2ca4c3bc835aab2d20d3920 to your computer and use it in GitHub Desktop.
modeltf = keras.Sequential([
keras.layers.Conv2D(input_shape=(28,28,1), filters=16, kernel_size=5, strides=1, padding="same", activation=tf.nn.relu),
keras.layers.BatchNormalization(),
keras.layers.MaxPooling2D(pool_size=2, strides=2),
keras.layers.Conv2D(32, kernel_size=5, strides=1, padding="same", activation=tf.nn.relu),
keras.layers.BatchNormalization(),
keras.layers.MaxPooling2D(pool_size=2, strides=2),
keras.layers.Flatten(),
keras.layers.Dense(10, activation=tf.nn.softmax)
])
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment