kl_loss function in Convolutional flipout layers

IntelLabs / bayesian-torch

A library for Bayesian neural network layers and uncertainty estimation in Deep Learning extending the core of PyTorch

BSD 3-Clause "New" or "Revised" License

544 stars 73 forks source link

Closed ranganathkrishnan closed 2 years ago

ranganathkrishnan commented 2 years ago

include kl_loss() function in Convolutional flipout layers, to compute kl when 'return_kl' flag is set to False.

ranganathkrishnan commented 2 years ago

Fixed with commit 97ebe28ce53ae902a5e40daaa5bc982db850da03