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

Smashing bugs (dimension mismatch between Normal and

Browse Source 
Independent/MultivariateNormal)
This commit is contained in:
dodox
2022-07-15 15:46:31 +02:00
parent ab557a8856
commit a86d19053d
4 changed files with 21 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files
metastable_baselines/distributions/distributions.py
+7 -5
View File
@@ -41,8 +41,8 @@ class ParametrizationType(Enum):
CHOL = 1
SPHERICAL_CHOL = 2
# Not (yet?) implemented:
#GIVENS = 3
#NNLN_EIGEN = 4
# GIVENS = 3
# NNLN_EIGEN = 4
class EnforcePositiveType(Enum):
@@ -360,13 +360,15 @@ class CholNet(nn.Module):
raise Exception()
def _chol_from_flat(self, flat_chol):
chol = fill_triangular(flat_chol).expand(self._flat_chol_len, -1, -1)
# chol = fill_triangular(flat_chol).expand(self._flat_chol_len, -1, -1)
chol = fill_triangular(flat_chol)
return self._ensure_diagonal_positive(chol)
def _chol_from_flat_sphe_chol(self, flat_sphe_chol):
pos_flat_sphe_chol = self._ensure_positive_func(flat_sphe_chol)
sphe_chol = fill_triangular(pos_flat_sphe_chol).expand(
self._flat_chol_len, -1, -1)
# sphe_chol = fill_triangular(pos_flat_sphe_chol).expand(
# self._flat_chol_len, -1, -1)
sphe_chol = fill_triangular(pos_flat_sphe_chol)
chol = self._chol_from_sphe_chol(sphe_chol)
return chol