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

Python package restructuring #44

Closed vakker closed 4 years ago

vakker commented 4 years ago

It would be great if this was an actual Python package that can be installed with pip install -e . It would require some restructuring and a setup.py, but no code changes. I don't know if it's something you wanted to do, but it would be beneficial for people using the library.

kumar-shridhar commented 4 years ago

Hi @vakker ,

We are making some final changes to the repo. Some measures about uncertainty estimation and a different initialization for the layers. It will take 2 weeks overall. After that, we will definitely put things inside a python package. Thanks for notifying us.