Commit 4e2ee1c2 authored by lucas_miranda's avatar lucas_miranda
Browse files

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

parent 2952a4df
Pipeline #100336 passed with stages
in 26 minutes and 33 seconds
......@@ -114,7 +114,7 @@ rule coarse_hyperparameter_tuning:
"--kl-warmup 30 "
"--mmd-warmup 30 "
"--encoding-size {wildcards.enc} "
"--batch-size 256 "
"--batch-size 512 "
"--window-size 24 "
"--window-step 12 "
"--output-path {outpath}coarse_hyperparameter_tuning "
......
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