Added experiment runner shs, also fixed variable name error.

csadrian · Apr 1, 2019 · 0b5c2a4 · 0b5c2a4
1 parent af71f43
commit 0b5c2a4
Show file tree

Hide file tree

Showing 5 changed files with 25 additions and 1 deletion.
diff --git a/a.sh b/a.sh
@@ -0,0 +1,6 @@
+
+for i in {0..15}
+do
+  CUDA_VISIBLE_DEVICES=0 dlib_reproduce --study=augmented_variance_vae_study_v1 --model_num=${i} > output/${i}.cout 2> output/${i}.cerr
+done
+
diff --git a/b.sh b/b.sh
@@ -0,0 +1,6 @@
+
+for i in {16..31}
+do
+  CUDA_VISIBLE_DEVICES=1 dlib_reproduce --study=augmented_variance_vae_study_v1 --model_num=${i} > output/${i}.cout 2> output/${i}.cerr
+done
+
diff --git a/c.sh b/c.sh
@@ -0,0 +1,6 @@
+
+for i in {32..48}
+do
+  CUDA_VISIBLE_DEVICES=2 dlib_reproduce --study=augmented_variance_vae_study_v1 --model_num=${i} > output/${i}.cout 2> output/${i}.cerr
+done
+
diff --git a/d.sh b/d.sh
@@ -0,0 +1,6 @@
+
+for i in {49..63}
+do
+  CUDA_VISIBLE_DEVICES=3 dlib_reproduce --study=augmented_variance_vae_study_v1 --model_num=${i} > output/${i}.cout 2> output/${i}.cerr
+done
+
diff --git a/disentanglement_lib/methods/unsupervised/balanced_vae.py b/disentanglement_lib/methods/unsupervised/balanced_vae.py
@@ -85,7 +85,7 @@ def model_fn(self, features, labels, mode, params):
           loss=loss,
           eval_metrics=(make_metric_fn("reconstruction_loss", "elbo",
                                        "regularizer", "kl_loss" ,"size_loss", "variance_loss", "regularizer_loss"),
-                        [reconstruction_loss, -elbo, regularizer, kl_loss, size_loss, variance_loss, regularizer_loss]))
+                        [reconstruction_loss, -elbo, regularizer, kl_loss, size_loss, variance_loss, regularizer]))
     else:
       raise NotImplementedError("Eval mode not supported.")