Pyro Discussion Forum
Aug 18, 2025 · Forum For Pyro Developers
Bad performance Bayesian Convolutional Neural Network - Misc.
Apr 28, 2025 · Hello Pyro community, I’m trying to build a Bayesian CNN for MNIST classification using Pyro, but despite seeing the ELBO loss decrease to around 10 during training, the …
numpyro - Pyro Discussion Forum
Jun 3, 2019 · Forum For Pyro Developers
Help post，shape problem - Pyro Discussion Forum
Apr 25, 2025 · with pyro.plate("data", x.size(0)): pyro.sample("obs", dist.Normal(mu, sigma), obs=y.squeeze(-1),infer={"scale": annealing_factor}) #obs，真实数据+噪声
Reducing MCMC memory usage - numpyro - Pyro Discussion Forum
Oct 15, 2023 · I am running NUTS/MCMC (on multiple CPU cores) for a quite large dataset (400k samples) for 4 chains x 2000 steps. mcmc.run actually ran until the end, but then died with an …
Implementation & normalizing flow in matrix normal distribution
Nov 1, 2024 · Hi, I’m working on a model where the likelihood follows a matrix normal distribution, X ~ MN_{n,p} (M, U, V). I’m using conjugate priors: M ~ MN U ~ Inverse Wishart V ~ Inverse …
GPU available but slow compared to CPU on BNN example
Nov 25, 2021 · Well, It is a VI + NeuraReparametrisation to perform a NUTS. Currently, we have found that the job runs 100% on CPU while GPU memory is activated. it’s as if the CPU was …
Multi GPU num_samples - Pyro Discussion Forum
Jan 10, 2021 · Hello I was wondering if it’s possible to parallelize the number of particles used to estimate the ELBO over multiple GPUs (i.e. num_particles=40, and if we have 4 GPUs, then …
How to reduce the memory usage? - numpyro - Pyro Discussion …
Jan 15, 2025 · Hello, I’m having a “Memory Exhausion” issue: Is there a way to reduce the usage of memory of the algorithm? (Some sort of batch_size?) Thanks in advance
Infer_discrete with multiple sites and enumeration - Misc. - Pyro ...
Apr 30, 2023 · I’m having trouble with a hierarchical mixture model, whose posterior inference is performed with autoguide and naive enumeration (I’m not confident about whether I’m …