result seems not good on Bosten example

[hangzhang23]

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)

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

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