Skip to content

Latest commit

 

History

History
13 lines (10 loc) · 282 Bytes

losses.md

File metadata and controls

13 lines (10 loc) · 282 Bytes
Raw

Losses

Index:

KL Divergence from nn

blitz.losses.kl_divergence_from_nn(model)

Returns the summed KL Divergence of each of the models bayesian layers.

Parameters:

  • model - torch.nn.Module