search

williamFalcon / pytorch-lightning-vae

VAE for color images
Apache License 2.0
207 stars 20 forks source link

Update image_plotting_callback.py #1

Open jordiclive opened 3 years ago

jordiclive commented 3 years ago

p = torch.distributions.Normal(torch.zeros_like(rand_v), torch.zeros_like(rand_v)) should be ones_like for variance