Bayesian Neural Network for PyTorch
Project description
Bayesian-Neural-Network-Pytorch
This is a lightweight repository of bayesian neural network for Pytorch. There are bayesian versions of pytorch layers and some utils. The aim is to help construct bayesian neural network intuitively.
Usage
Dependencies
- torch 1.2.0
- python 3.6
Installation
pip install torchbnn
orgit clone https://github.com/Harry24k/bayesian-neural-network-pytorch
import torchbnn
Thanks to
- @kumar-shridhar github:PyTorch-BayesianCNN
- @xuanqing94 github:BayesianDefense
Update Records
Version 0.1
- modules : BayesLinear, BayesConv2d, BayesBatchNorm2d
- utils : convert_model(nonbayes_to_bayes, bayes_to_nonbayes)
- functional : bayesian_kl_loss
Project details
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distributions
No source distribution files available for this release.See tutorial on generating distribution archives.
Built Distribution
torchbnn-0.1-py3-none-any.whl
(8.0 kB
view hashes)