search

AdamCobb / hamiltorch

PyTorch-based library for Riemannian Manifold Hamiltonian Monte Carlo (RMHMC) and inference in Bayesian neural networks
BSD 2-Clause "Simplified" License
426 stars 63 forks source link

Fix shapes of the zero-means when passing taus into the model #24

Closed asolin closed 1 year ago

asolin commented 1 year ago

Bugfix: For passing taus into the model, the mean and std of the Gaussian need to have matching shapes.

AdamCobb commented 1 year ago

Thanks!