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
index 358ebb002..546502361 100644
--- 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