From 2f365a48257a798fe0d348ee632bae2edf528f9d Mon Sep 17 00:00:00 2001 From: lucas_miranda <lucasmiranda42@gmail.com> Date: Wed, 19 May 2021 19:42:35 +0200 Subject: [PATCH] Replaced for loop with vectorised mapping on ClusterOverlap regularization layer --- deepof_experiments.smk | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepof_experiments.smk b/deepof_experiments.smk index 09e9d724..4fa0e547 100644 --- a/deepof_experiments.smk +++ b/deepof_experiments.smk @@ -157,7 +157,7 @@ rule train_models: "--rule-based-prediction {wildcards.rulesweight} " "--latent-reg {wildcards.latreg} " "--loss {wildcards.loss} " - "--overlap_loss {wildcards.overlap_loss} " + "--overlap-loss {wildcards.overlap_loss} " "--kl-annealing-mode {wildcards.warmup_mode} " "--kl-warmup {wildcards.warmup} " "--mmd-annealing-mode {wildcards.warmup_mode} " -- GitLab