Harry24k / bayesian-neural-network-pytorch

PyTorch implementation of bayesian neural network [torchbnn]
MIT License
496 stars 76 forks source link

posterior_mu, posterior_sigma, W_sigma #7

Open starsky68 opened 4 years ago

starsky68 commented 4 years ago

https://github.com/kumar-shridhar/PyTorch-BayesianCNN/blob/81e6a354e2638cfad488a139dea4a432d9ad3578/layers/BBB/BBBLinear.py#L27 https://github.com/kumar-shridhar/PyTorch-BayesianCNN/blob/81e6a354e2638cfad488a139dea4a432d9ad3578/layers/BBB/BBBLinear.py#L57

The above two links seem to be a bit different from the implementation in this open source project. I have some things that I don’t understand. Can you explain it?