Skip to content
Snippets Groups Projects

Fix typo in TF docs

Merged Jakub Beránek requested to merge beranekj/docs.it4i.cz:tf-typo into master
1 file
+ 1
1
Compare changes
  • Side-by-side
  • Inline
@@ -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)),
Loading