Skip to content
Snippets Groups Projects
Commit a91a6400 authored by Jan Siwiec's avatar Jan Siwiec
Browse files

Merge branch 'tf-typo' into 'master'

Fix typo in TF docs

See merge request !416
parents 51870c71 445ec57b
No related branches found
No related tags found
1 merge request!416Fix typo in TF docs
Pipeline #29174 passed with warnings
...@@ -112,7 +112,7 @@ train_dataset = mnist_train.map(scale).cache().shuffle(BUFFER_SIZE).batch(BATCH_ ...@@ -112,7 +112,7 @@ train_dataset = mnist_train.map(scale).cache().shuffle(BUFFER_SIZE).batch(BATCH_
eval_dataset = mnist_test.map(scale).batch(BATCH_SIZE) eval_dataset = mnist_test.map(scale).batch(BATCH_SIZE)
# The following line makes sure that the model will run on multiple GPUs (if they are available) # The following line makes sure that the model will run on multiple GPUs (if they are available)
# Without `strategy.scopy()`, the model would only be trained on a single GPU # Without `strategy.scope()`, the model would only be trained on a single GPU
with strategy.scope(): with strategy.scope():
model = tf.keras.Sequential([ model = tf.keras.Sequential([
tf.keras.layers.Conv2D(32, 3, activation='relu', input_shape=(28, 28, 1)), tf.keras.layers.Conv2D(32, 3, activation='relu', input_shape=(28, 28, 1)),
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment