Fixed Bug: _chol_from_sphe_chol dependet on action_dim, not able to use

for Hybrid methods
2022-07-16 15:47:09 +02:00 · 2022-07-16 15:47:09 +02:00 · c141599662
commit c141599662
parent 3fa6de7e66
1 changed files with 1 additions and 1 deletions
--- a/metastable_baselines/distributions/distributions.py
+++ b/metastable_baselines/distributions/distributions.py
@ -389,7 +389,7 @@ class CholNet(nn.Module):
        # We already ensure S > 0 in _chol_from_flat_sphe_chol
        # We ensure < pi by applying tanh*pi to all applicable elements
        S = sphe_chol
-        n = self.action_dim
+        n = sphe_chol.shape[-1]
        L = th.zeros_like(sphe_chol)
        for i in range(n):
            t = 1