pytorch-lightning-vae
pytorch-lightning-vae copied to clipboard
Update image_plotting_callback.py
p = torch.distributions.Normal(torch.zeros_like(rand_v), torch.zeros_like(rand_v)) should be ones_like for variance