Closed unnir closed 3 years ago
Hello @unnir and thank you so much for using BLiTZ.
For each layer, you can get the rho for weight and bias attributes and follow the equataion from the paper:
log (1 + exp(rho))
The resulting matrix will correpspond to the std you are looking for
Best regards, -Pi
Closing due to staleness.
Hey,
sorry, I could not find the answer.
How can I get the uncertainty estimate for each weight in the BayesianLinear layer? I mean the std for each weighty.
Thanks