From bc0e188a0de105f6587d1887883a1993e0d17fff Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Sat, 16 Jul 2022 15:19:56 +0200 Subject: [PATCH] Removed debugging-code --- 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 9d63bc8..82c1dbc 100644 --- a/metastable_baselines/distributions/distributions.py +++ b/metastable_baselines/distributions/distributions.py @@ -346,11 +346,7 @@ class CholNet(nn.Module): pearson_cor = pearson_cor.expand( (stds.shape[0],)+pearson_cor.shape) stds = stds.unsqueeze(2) - try: - cov = stds.mT * pearson_cor * stds - except: - import pdb - pdb.set_trace() + cov = stds.mT * pearson_cor * stds chol = th.linalg.cholesky(cov) return chol elif self.par_strength == Strength.SCALAR and self.cov_strength == Strength.FULL: