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

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

Browse Source 
for Hybrid methods
This commit is contained in:
Dominik Moritz Roth 2022-07-16 15:47:09 +02:00
parent 3fa6de7e66
commit c141599662
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):
# We already ensure S > 0 in _chol_from_flat_sphe_chol # We already ensure S > 0 in _chol_from_flat_sphe_chol
# We ensure < pi by applying tanh*pi to all applicable elements # We ensure < pi by applying tanh*pi to all applicable elements
S = sphe_chol S = sphe_chol
n = self.action_dim n = sphe_chol.shape[-1]
L = th.zeros_like(sphe_chol) L = th.zeros_like(sphe_chol)
for i in range(n): for i in range(n):
t = 1 t = 1