Skip to main content

Dynamic VAE Layer Creation

1 min

What I worked on

Created a function to build encoder and decoder layers dynamically using input_dim, latent_dim, num_layers, and scale_factor. Refactored redundant code between encoder and decoder builders.

What I noticed

  • Hidden dims can scale down each layer by a factor

”Aha” Moment

na

What still feels messy

n/a

Next step

Use the dynamic builder to easily test different VAE depths and scaling strategies.