diff --git a/docs.it4i/software/machine-learning/tensorflow.md b/docs.it4i/software/machine-learning/tensorflow.md index 8e7386269a8c778331b87eb40d1d45da22adc678..7a8cfb7d70154f087b754e0337dc0101945f1017 100644 --- a/docs.it4i/software/machine-learning/tensorflow.md +++ b/docs.it4i/software/machine-learning/tensorflow.md @@ -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) # 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(): model = tf.keras.Sequential([ tf.keras.layers.Conv2D(32, 3, activation='relu', input_shape=(28, 28, 1)),