Make mlp_probe_per_layer deediff's default

daniel-gallo · daniel-gallo · commit 9e94a4286503 · 2024-07-27T16:06:25.000+02:00
diff --git a/configs/deediff_celeba.yaml b/configs/deediff_celeba.yaml
@@ -10,4 +10,4 @@ model_params:
   mlp_time_embed: False
   num_classes: -1
   normalize_timesteps: True
-  classifier_type: "attention_probe"
+  classifier_type: "mlp_probe_per_layer"
diff --git a/configs/deediff_cifar10.yaml b/configs/deediff_cifar10.yaml
@@ -10,4 +10,4 @@ model_params:
   mlp_time_embed: False
   num_classes: -1
   normalize_timesteps: True
-  classifier_type: "attention_probe"
+  classifier_type: "mlp_probe_per_layer"
