cond module updates

DAI-Lab · Dec 20, 2024 · bdf831d · bdf831d
1 parent 9c8a6bb
commit bdf831d
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/config/config.yaml b/config/config.yaml
@@ -4,7 +4,7 @@ defaults:
   - evaluator: default
   - _self_
 
-device: 0 # 0, cpu
+device: 1 # 0, cpu
 
 job_name: ${model.name}_${dataset.name}_${dataset.user_group}
 run_dir: outputs/${job_name}/${now:%Y-%m-%d_%H-%M-%S}

diff --git a/config/model/acgan.yaml b/config/model/acgan.yaml
@@ -3,11 +3,11 @@ name: acgan
 noise_dim: 256
 cond_emb_dim: 64
 sparse_conditioning_loss_weight: 0.5
-kl_weight: 0.1
+kl_weight: 0.5
 
-batch_size: 32
+batch_size: 64
 n_epochs: 200
-lr_gen: 3e-4
+lr_gen: 2e-4
 lr_discr: 1e-4
 warm_up_epochs: 100
 include_auxiliary_losses: True

diff --git a/eval/evaluator.py b/eval/evaluator.py
@@ -263,7 +263,7 @@ def create_visualizations(
         dataset: Any,
         model: Any,
         num_samples: int = 100,
-        num_runs: int = 10,
+        num_runs: int = 3,
     ):
         """
         Create various visualizations for the evaluation results.

diff --git a/generator/conditioning.py b/generator/conditioning.py
@@ -3,7 +3,7 @@
 
 
 class ConditioningModule(nn.Module):
-    def __init__(self, categorical_dims, embedding_dim, device, alpha=0.8):
+    def __init__(self, categorical_dims, embedding_dim, device, alpha=0.1):
         super(ConditioningModule, self).__init__()
         self.embedding_dim = embedding_dim
         self.device = device