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

Removed debug-point

Browse Source
This commit is contained in:
Dominik Moritz Roth 2022-07-16 14:58:29 +02:00
parent 4854346f2d
commit 04529e8261
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
metastable_baselines/distributions/distributions.py
View File

@ -338,11 +338,7 @@ class CholNet(nn.Module):
stds = self._ensure_positive_func(self.stds(x)) stds = self._ensure_positive_func(self.stds(x))
smol = self._parameterize_full(self.params) smol = self._parameterize_full(self.params)
big = self.padder(smol) big = self.padder(smol)
try:
pearson_cor_chol = big + th.eye(stds.shape[-1]) pearson_cor_chol = big + th.eye(stds.shape[-1])
except:
import pdb
pdb.set_trace()
pearson_cor = pearson_cor_chol.T @ pearson_cor_chol pearson_cor = pearson_cor_chol.T @ pearson_cor_chol
cov = stds.T * pearson_cor * stds cov = stds.T * pearson_cor * stds
chol = th.linalg.cholesky(cov) chol = th.linalg.cholesky(cov)