fixed config bug

DAI-Lab · Oct 17, 2024 · 0767e03 · 0767e03
1 parent 888d05b
commit 0767e03
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/config/model_config.yaml b/config/model_config.yaml
@@ -1,4 +1,4 @@
-device: 1 # 0, cpu
+device: 0 # 0, cpu
 seq_len: 96 # should not be changed for the current datasets
 input_dim: 1 # or 1 depending on user, but is dynamically set
 noise_dim: 256
@@ -74,4 +74,5 @@ acgan:
   lr_gen: 3e-4
   lr_discr: 1e-4
   warm_up_epochs: 100
-  include_auxiliary_losses: True
+  include_auxiliary_losses: False
+  save_cycle: 100
diff --git a/main.py b/main.py
@@ -47,7 +47,7 @@ def main():
     # evaluate_individual_user_models("acgan", include_generation=True)
     # evaluate_individual_user_models("acgan", include_generation=False, normalization_method="date")
     evaluate_single_dataset_model(
-        "diffusion_ts",
+        "acgan",
         geography="california",
         include_generation=True,
         normalization_method="group",