YannDubs / disentangling-vae

Experiments for understanding disentanglement in VAE latent representations
Other
785 stars 143 forks source link

PlotNeuralNet Code #65

Closed ericluo04 closed 3 years ago

ericluo04 commented 3 years ago

Hello! I was kindly wondering if you could share your PlotNeuralNet code (which is what I presume you used to generate this). I would really appreciate it! To my knowledge, there aren't any VAE visualization examples for this online, so I imagine it would be very helpful for others too.

linesd commented 3 years ago

Hello, indeed PlotNeuralNet was used to generate the architecture diagram. I checked but unfortunately I deleted the code to produce it - but if I remember correctly it was straight forward and mainly used already existing blocks from PlotNeuralNet with minor adjustments.

ericluo04 commented 3 years ago

Ohh no problem - thanks for checking anyways!