From 3110275d7b83c7e07da42d719b60088a97591604 Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Fri, 7 Oct 2022 11:38:34 +0200 Subject: [PATCH] Fixed bug with EIGEN (par_type) --- metastable_baselines/distributions/distributions.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/metastable_baselines/distributions/distributions.py b/metastable_baselines/distributions/distributions.py index 951eca4..282d69c 100644 --- a/metastable_baselines/distributions/distributions.py +++ b/metastable_baselines/distributions/distributions.py @@ -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)