Bayesian regression nnet

The call to pyro.random_module lifts the parameters in nn.Module to random variables.

This forum post will be helpful.

2 Likes