piEsposito / blitz-bayesian-deep-learning

A simple and extensible library to create Bayesian Neural Network layers on PyTorch.
GNU General Public License v3.0
921 stars 106 forks source link

fix: .normal_() on self.eps_weight.data #21

Closed ana-tamais closed 4 years ago

ana-tamais commented 4 years ago

Now it works with siamese architectures and when we need to double feedforward before gathering loss and backpropagating.