Simplifying precomputed labels structure

SewoongLab · Jan 3, 2024 · 75c921e · 75c921e
1 parent 4661e36
commit 75c921e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 9 deletions.
diff --git a/experiments/example_precomputed/config.toml b/experiments/example_precomputed/config.toml
@@ -1,15 +1,11 @@
 [downstream]
-input = "precomputed/cifar/r32p/1xs/1500/labels.npy"
-downstream_model = "vit-pretrain"
+input = "precomputed/cifar/r32p/1xs/1500.npy"
+downstream_model = "r32p"
 trainer = "sgd"
 dataset = "cifar"
 source_label = 9
 target_label = 4
 poisoner = "1xs"
 output_path = "experiments/example_precomputed/"
 soft = false
-alpha = 0.0
-
-[downstream.optim_kwargs]
-lr = 0.01
-weight_decay = 0.0002
+alpha = 0.0
diff --git a/experiments/example_precomputed_mix/config.toml b/experiments/example_precomputed_mix/config.toml
@@ -1,12 +1,12 @@
 [downstream]
-input = "precomputed/cifar/r32p/1xs/1500/labels.npy"
+input = "precomputed/cifar/r32p/1xs/1500.npy"
 downstream_model = "vit-pretrain"
 trainer = "sgd"
 dataset = "cifar"
 source_label = 9
 target_label = 4
 poisoner = "1xs"
-output_path = "experiments/example_precomputed/"
+output_path = "experiments/example_precomputed_mix/"
 soft = false
 alpha = 0.0