dodox/metastable-baselines
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Removed debugging-code

Browse Source
This commit is contained in:
Dominik Moritz Roth 2022-07-16 15:19:56 +02:00
parent d2d84d3287
commit bc0e188a0d
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
metastable_baselines/distributions/distributions.py
View File

@ -346,11 +346,7 @@ class CholNet(nn.Module):
pearson_cor = pearson_cor.expand( pearson_cor = pearson_cor.expand(
(stds.shape[0],)+pearson_cor.shape) (stds.shape[0],)+pearson_cor.shape)
stds = stds.unsqueeze(2) stds = stds.unsqueeze(2)
try: cov = stds.mT * pearson_cor * stds
cov = stds.mT * pearson_cor * stds
except:
import pdb
pdb.set_trace()
chol = th.linalg.cholesky(cov) chol = th.linalg.cholesky(cov)
return chol return chol
elif self.par_strength == Strength.SCALAR and self.cov_strength == Strength.FULL: elif self.par_strength == Strength.SCALAR and self.cov_strength == Strength.FULL: