From 04529e8261a310d4e18532432802611c9465467e Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Sat, 16 Jul 2022 14:58:29 +0200 Subject: [PATCH] Removed debug-point --- metastable_baselines/distributions/distributions.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/metastable_baselines/distributions/distributions.py b/metastable_baselines/distributions/distributions.py index 14f8fc2..d4742fb 100644 --- a/metastable_baselines/distributions/distributions.py +++ b/metastable_baselines/distributions/distributions.py @@ -338,11 +338,7 @@ class CholNet(nn.Module): stds = self._ensure_positive_func(self.stds(x)) smol = self._parameterize_full(self.params) big = self.padder(smol) - try: - pearson_cor_chol = big + th.eye(stds.shape[-1]) - except: - import pdb - pdb.set_trace() + pearson_cor_chol = big + th.eye(stds.shape[-1]) pearson_cor = pearson_cor_chol.T @ pearson_cor_chol cov = stds.T * pearson_cor * stds chol = th.linalg.cholesky(cov)