Commit 2f365a48 authored by lucas_miranda's avatar lucas_miranda
Browse files

Replaced for loop with vectorised mapping on ClusterOverlap regularization layer

parent 19d722aa
Pipeline #101812 passed with stages
in 22 minutes and 57 seconds
...@@ -157,7 +157,7 @@ rule train_models: ...@@ -157,7 +157,7 @@ rule train_models:
"--rule-based-prediction {wildcards.rulesweight} " "--rule-based-prediction {wildcards.rulesweight} "
"--latent-reg {wildcards.latreg} " "--latent-reg {wildcards.latreg} "
"--loss {wildcards.loss} " "--loss {wildcards.loss} "
"--overlap_loss {wildcards.overlap_loss} " "--overlap-loss {wildcards.overlap_loss} "
"--kl-annealing-mode {wildcards.warmup_mode} " "--kl-annealing-mode {wildcards.warmup_mode} "
"--kl-warmup {wildcards.warmup} " "--kl-warmup {wildcards.warmup} "
"--mmd-annealing-mode {wildcards.warmup_mode} " "--mmd-annealing-mode {wildcards.warmup_mode} "
......
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