Bugs in main_Bayes.py - Githubissues

kumar-shridhar / PyTorch-BayesianCNN

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

MIT License

1.43k stars 325 forks source link

Bugs in main_Bayes.py #22

Closed jiahaosu closed 5 years ago

jiahaosu commented 5 years ago

Hello,

I tried to run a Bayesian LeNet5 on MNIST. However, I have difficulties in running the main_Bayes.py. It seems to me there are multiple bugs in the files (Flatten layer is not defined, arguments without defaults are defined after the ones that have defaults, a string is called as functions, etc). Could you let me know how I can call your code properly?

Best

kumar-shridhar commented 5 years ago

Hello @jiahaosu ,

Please run the ipython notebook Bayesian_CNN_Detailed.ipynb. Everything works there. If you want to run LeNet, please see the AlexNet file and it works. Just use the same structure. It should work.

Kumar