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.

Examples using `probly.train.bayesian.torch.collect_kl_divergence`¶

Credal BNN Visualization

Credal BNN Visualization

Credal BNN on MNIST

Credal BNN on MNIST

Bayesian Neural Network on Two Moons

Bayesian Neural Network on Two Moons

Bayesian Ensemble on Two Moons

Bayesian Ensemble on Two Moons

Bayesian Ensemble on MNIST

Bayesian Ensemble on MNIST

Bayesian Neural Network on MNIST

Bayesian Neural Network on MNIST