Remove autoencoder from the config file

razvanmatisan · razvanmatisan · commit e811f76d3231 · 2024-09-06T20:25:19.000+02:00
diff --git a/configs/deediff_imagenet64.yaml b/configs/deediff_imagenet64.yaml
@@ -10,7 +10,4 @@ model_params:
   mlp_time_embed: False
   num_classes: 1000
   normalize_timesteps: False
-  classifier_type: "mlp_probe_per_layer"
-
-autoencoder:
-  autoencoder_checkpoint_path: ./checkpoints/autoencoder/autoencoder_kl.pth
+  classifier_type: "mlp_probe_per_layer"
