probly.train.bayesian.torch.collect_kl_divergence¶
- probly.train.bayesian.torch.collect_kl_divergence(model: Predictor) torch.Tensor[source]¶
Collect the KL divergence of the Bayesian model by summing the KL divergence of each Bayesian layer.
Back to top