Commit 7e69c2f7 authored by lucas_miranda's avatar lucas_miranda
Browse files

Added a MirroredStrategy to train models on multiple GPUs if they are available

parent 44e98b2c
Pipeline #100314 passed with stages
in 32 minutes and 52 seconds
......@@ -326,7 +326,7 @@ def autoencoder_fitting(
X_val_dataset = (
tf.data.Dataset.from_tensor_slices(X_val)
.with_options(options)
.batch(batch_size)
.batch(batch_size * strategy.num_replicas_in_sync)
)
# Defines what to log on tensorboard (useful for trying out different models)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment