From 116265bb25c09dd05ba5060e1bbb97a1e9ce074e Mon Sep 17 00:00:00 2001 From: Lionel Peer Date: Sun, 8 Oct 2023 16:01:12 +0200 Subject: [PATCH] fixing docs build --- diffusion_models/losses/kl_divergence.py | 4 ---- 1 file changed, 4 deletions(-) diff --git a/diffusion_models/losses/kl_divergence.py b/diffusion_models/losses/kl_divergence.py index 395a27f..60ec0d6 100644 --- a/diffusion_models/losses/kl_divergence.py +++ b/diffusion_models/losses/kl_divergence.py @@ -11,8 +11,6 @@ def gaussian_kl( """Calculate KL Divergence of 2 Gaussian distributions. KL divergence between two univariate Gaussians, as derived in [1], with k=1 (dimensionality). - .. math:: - D_{KL}(p||q) = \frac{1}{2}\left[\log\frac{|\Sigma_q|}{|\Sigma_p|} - k + (\boldsymbol{\mu_p}-\boldsymbol{\mu_q})^T\Sigma_q^{-1}(\boldsymbol{\mu_p}-\boldsymbol{\mu_q}) + tr\left\{\Sigma_q^{-1}\Sigma_p\right\}\right] Parameters ---------- @@ -45,8 +43,6 @@ def log_gaussian_kl( """Calculate KL Divergence of 2 Gaussian distributions. KL divergence between two univariate Gaussians, as derived in [1], with k=1 (dimensionality) and log variances. - .. math:: - D_{KL}(p||q) = \frac{1}{2}\left[\log\frac{|\Sigma_q|}{|\Sigma_p|} - k + (\boldsymbol{\mu_p}-\boldsymbol{\mu_q})^T\Sigma_q^{-1}(\boldsymbol{\mu_p}-\boldsymbol{\mu_q}) + tr\left\{\Sigma_q^{-1}\Sigma_p\right\}\right] Parameters ----------