search

piEsposito / blitz-bayesian-deep-learning

A simple and extensible library to create Bayesian Neural Network layers on PyTorch.
GNU General Public License v3.0
918 stars 107 forks source link

result seems not good on Bosten example #101

Open hangzhang23 opened 2 years ago

hangzhang23 commented 2 years ago

Hi, I try to implement the Boston example and use a plot to see fitting result. but the result seems not good enough.

The followings are train result and validation result (mean with 3 times std) train_bosten_result_120 val_bosten_result_120

So I print the criterion from elbo function, but the mse curve seems like a oscillation, not a expected descending trend. bosten_loss_120

I am curious why mse is not descending? looking for your replying :)