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
### Bayesian layers and utilities to perform stochastic variational inference in PyTorch
5
5
@@ -8,14 +8,14 @@ Bayesian-Torch is designed to be flexible and seamless in extending a determinis
8
8
9
9
10
10
The repository has implementations for the following Bayesian layers:
11
-
-[x]**[Variational layers with reparameterized Monte Carlo estimators](bayesian_torch/layers/variational_layers)**[[Blundell et al. 2015](https://arxiv.org/abs/1505.05424)]
11
+
-[x]**[Variational layers with reparameterized Monte Carlo estimators](https://github.com/IntelLabs/bayesian-torch/tree/main/bayesian_torch/layers/variational_layers)**[[Blundell et al. 2015](https://arxiv.org/abs/1505.05424)]
-[x]**[Variational layers with Flipout Monte Carlo estimators](bayesian_torch/layers/flipout_layers)**[[Wen et al. 2018](https://arxiv.org/abs/1803.04386)]
18
+
-[x]**[Variational layers with Flipout Monte Carlo estimators](https://github.com/IntelLabs/bayesian-torch/tree/main/bayesian_torch/layers/flipout_layers)**[[Wen et al. 2018](https://arxiv.org/abs/1803.04386)]
@@ -35,7 +35,9 @@ The repository has implementations for the following Bayesian layers:
35
35
LSTMMixture
36
36
-->
37
37
38
+
<!--
38
39
Please refer to [documentation](doc/bayesian_torch.layers.md#layers) of Bayesian layers for details.
40
+
-->
39
41
40
42
Other features include:
41
43
-[x][dnn_to_bnn()](https://github.com/IntelLabs/bayesian-torch/blob/main/bayesian_torch/models/dnn_to_bnn.py#L127): An API to convert deterministic deep neural network (dnn) model of any architecture to Bayesian deep neural network (bnn) model, simplifying the model definition i.e. drop-in replacements of Convolutional, Linear and LSTM layers to corresponding Bayesian layers. This will enable seamless conversion of existing topology of larger models to Bayesian deep neural network models for extending towards uncertainty-aware applications.
(2) For building custom models, we have provided [example model implementations](bayesian_torch/models/bayesian) using the Bayesian layers.
145
+
(2) For building custom models, we have provided [example model implementations](https://github.com/IntelLabs/bayesian-torch/tree/main/bayesian_torch/models/bayesian) using the Bayesian layers.
144
146
145
147
## Example usage (training and evaluation of models)
0 commit comments