kl_loss function in Convolutional flipout layers
ranganathkrishnan opened this issue · comments
include kl_loss() function in Convolutional flipout layers, to compute kl when 'return_kl' flag is set to False.
Fixed with commit 97ebe28
IntelLabs / bayesian-torch
A library for Bayesian neural network layers and uncertainty estimation in Deep Learning extending the core of PyTorch