Commit 9adbeaf5 authored by lucas_miranda's avatar lucas_miranda
Browse files

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

parent 4e2ee1c2
Pipeline #100342 passed with stages
in 24 minutes and 46 seconds
......@@ -443,7 +443,7 @@ def autoencoder_fitting(
+ [
CustomStopper(
monitor="val_loss",
patience=10,
patience=15,
restore_best_weights=True,
start_epoch=max(kl_warmup, mmd_warmup),
),
......@@ -467,7 +467,7 @@ def autoencoder_fitting(
callbacks_ = cbacks + [
CustomStopper(
monitor="val_loss",
patience=5,
patience=15,
restore_best_weights=True,
start_epoch=max(kl_warmup, mmd_warmup),
),
......
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