Start assembling a nice configuration for ACS

wearepal · Feb 29, 2024 · 6e6002b · 6e6002b
1 parent 3553c4f
commit 6e6002b
Show file tree

Hide file tree

Showing 6 changed files with 34 additions and 7 deletions.
diff --git a/external_confs/dm/acs.yaml b/external_confs/dm/acs.yaml
@@ -0,0 +1,4 @@
+stratified_sampler: approx_group
+num_workers: 1
+batch_size_tr: 128
+batch_size_te: 100000
diff --git a/external_confs/ds/acs/employment_dis_fl.yaml b/external_confs/ds/acs/employment_dis_fl.yaml
@@ -7,4 +7,4 @@ survey_year: YEAR_2018
 states:
   - FL
 survey: PERSON
-horizon: ONE_YEAR
+horizon: ONE_YEAR
diff --git a/external_confs/experiment/acs/fcn.yaml b/external_confs/experiment/acs/fcn.yaml
@@ -0,0 +1,23 @@
+# @package _global_
+
+defaults:
+    - override /dm: acs
+    - override /ds: acs/employment_dis_fl
+    - override /split: acs/employment_dis_fl
+    - override /labeller: gt
+
+alg:
+    use_amp: False
+    steps: 30000
+    ga_steps: 1
+    max_grad_norm: null
+
+ae:
+    recon_loss: l2
+    lr: 1.e-4
+    optimizer_cls: ADAM
+    weight_decay: 0
+    zs_dim: 1
+
+eval:
+    batch_size: 100000
diff --git a/external_confs/split/acs/employment_dis.yaml b/external_confs/split/acs/employment_dis.yaml
@@ -5,4 +5,4 @@ defaults:
 seed: 0
 train_props:
   1:
-    0: 0.0
+    0: 0.0
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -30,7 +30,7 @@ scikit-image = ">=0.14"
 scikit_learn = { version = ">=0.20.1" }
 scipy = { version = ">=1.2.1" }
 seaborn = { version = ">=0.9.0" }
-torch-conduit = { version = ">=0.3.4", extras = ["image", "fair"] }
+torch-conduit = { version = ">=0.4.2", extras = ["image", "fair"] }
 typing_extensions = ">= 4.10"
 
 tqdm = { version = ">=4.31.1" }
-Original file line number
+Diff line change
@@ Expand Up / @@ -5,4 +5,4 @@ defaults: @@
     seed: 0
     train_props:
 :
-: 0.0
+: 0.0