You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Copy file name to clipboardExpand all lines: README.md
+6-1Lines changed: 6 additions & 1 deletion
Display the source diff
Display the rich diff
Original file line number
Diff line number
Diff line change
@@ -4,6 +4,8 @@ This is a lightweight repository of bayesian neural network for Pytorch.
4
4
There are bayesian versions of pytorch layers and some utils.
5
5
To help construct bayesian neural network intuitively, all codes are modified based on the original pytorch codes.
6
6
7
+
** Please use torch v1.2 **
8
+
7
9
## Usage
8
10
9
11
### Dependencies
@@ -91,4 +93,7 @@ To freeze a bayesian neural network, which means force a bayesian neural network
91
93
***BayesConv2d** : Bias will set to False if the bias in args is None or Flase. Otherwise, it set to True. In addition, re-defined with prior_sigma instead of prior_log_sigma.
92
94
93
95
***utils/convert_model.py** :
94
-
* Depreciated. Please refer the [modified demo](https://github.com/Harry24k/bayesian-neural-network-pytorch/blob/master/demos/Convert%20to%20Bayesian%20Neural%20Network.ipynb).
96
+
* Depreciated. Please refer the [modified demo](https://github.com/Harry24k/bayesian-neural-network-pytorch/blob/master/demos/Convert%20to%20Bayesian%20Neural%20Network.ipynb).
0 commit comments