{ "image_channels": 3, "latent_dim": 256, "hidden_dims": [ 32, 64, 128, 256 ], "image_width": 256, "image_height": 256, "kernel_size": [ 4, 4 ], "strides": [ 2, 2 ], "padding": "VALID", "dropout": 0.1, "conv_bias": true, "activation": "relu", "kernel_init": "lecun", "use_latent_bias": true, "use_batch_norm": true, "model_type": "autoencoder" }