Merge branch 'main' of github.com:razvanmatisan/early-stopping-diffusion

Alejandro Monroy Muñoz · Alejandro Monroy Muñoz · commit 8b560bac5685 · 2024-09-09T18:41:09.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"
