Implemented Minibatch Weighting

danielkelshaw / WeightUncertainty

PyTorch implementation of 'Weight Uncertainty in Neural Networks'

https://arxiv.org/abs/1505.05424

MIT License

18 stars 2 forks source link

Closed danielkelshaw closed 4 years ago

danielkelshaw commented 4 years ago

Provided an implementation of minibatch weighting according to the formula given in section 3.4 of the 'Weight Uncertainty in Neural Networks' paper.

Resolves: #13