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 = (
.batch(batch_size * strategy.num_replicas_in_sync)
# Defines what to log on tensorboard (useful for trying out different models)
Supports Markdown
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