configs

piercus · Feb 15, 2024 · 51b76fd · 51b76fd
1 parent 7b72432
commit 51b76fd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/configs/histogram-auto-encoder/train-histogram-autoencoder-4_16-32-b8-lr5e-3-kl_div.toml b/configs/histogram-auto-encoder/train-histogram-autoencoder-4_16-32-b8-lr5e-3-kl_div.toml
@@ -8,7 +8,7 @@ tags = ["autoencoder", "4bits"]
 
 [histogram_auto_encoder]
 latent_dim = 4
-resnet_sizes = [2, 2, 2, 4]
+resnet_sizes = [32, 32, 32, 32]
 n_down_samples = 3
 color_bits = 4
 num_groups = 1