About BBBlayers - Githubissues

kumar-shridhar / PyTorch-BayesianCNN

Bayesian Convolutional Neural Network with Variational Inference based on Bayes by Backprop in PyTorch.

MIT License

1.42k stars 323 forks source link

Closed Cherishyt closed 4 years ago

Cherishyt commented 4 years ago

There is some bug in BBBlayers.py. What is the initial value of p_logvar_init and q_logvar_init? They can't be set None.

kumar-shridhar commented 4 years ago

Hi, The values are initialized inside the Network (like AlexNet). Link here.

Cherishyt commented 4 years ago

Thanks! But the values haven't been initialized in LeNet code. Is it the same as in AlexNet?

kumar-shridhar commented 4 years ago

For now please make the LeNet code similar to AlexNet and it will work.