The code for the boston regression example in readme doesn't converge. It seems to differ from `blitz/examples/bayesian_regression_boston.py' a little bit. I changed an optimizer and added the activation function to match the code from examples and now the readme version converges as well.
There are other differences, like a number of epochs (100 vs 1000), cuda usage, and the complexity_cost_weight argument in elbo, but as the readme started to converge anyway, I've left out these changes to keep it simple.
The code for the boston regression example in readme doesn't converge. It seems to differ from `blitz/examples/bayesian_regression_boston.py' a little bit. I changed an optimizer and added the activation function to match the code from examples and now the readme version converges as well.
There are other differences, like a number of epochs (100 vs 1000), cuda usage, and the
complexity_cost_weight
argument in elbo, but as the readme started to converge anyway, I've left out these changes to keep it simple.