Skip to content
Snippets Groups Projects
Unverified Commit 445ec57b authored by Jakub Beránek's avatar Jakub Beránek
Browse files

Fix typo in TF docs

parent 83e4a5a6
Branches
Tags v2.0
1 merge request!416Fix typo in TF docs
...@@ -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