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

Fixed bug with EIGEN (par_type)

Browse Source
This commit is contained in:
Dominik Moritz Roth 2022-10-07 11:38:34 +02:00
parent f37d3215a6
commit 3110275d7b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
metastable_baselines/distributions/distributions.py
View File

@ -389,7 +389,7 @@ class CholNet(nn.Module):
self._full_params_len = self._flat_chol_len
elif self.par_type == ParametrizationType.EIGEN:
self._full_params_len = self.action_dim * 2
elif self.par_type == ParametrizationType.EIGEN_BIJECT:
elif self.par_type == ParametrizationType.EIGEN_RAW:
self._full_params_len = self.action_dim * 2
self._givens_rotator = givens.Rotation(action_dim)