saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-05-22 22:10:16 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Remove note

Browse source
This commit is contained in:
Antonin RAFFIN 2019-09-22 21:33:35 +02:00
parent 1bd2e42995
commit 98e9560913
1 changed files with 0 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
torchy_baselines/common/distributions.py
View file

@ -103,8 +103,6 @@ class SquashedDiagGaussianDistribution(DiagGaussianDistribution):
# Inverse tanh
# Naive implementation (not stable): 0.5 * torch.log((1 + x ) / (1 - x))
# We use numpy to avoid numerical instability
# TODO: store the gaussian action because of the action added
# this would avoid also inverting the tanh
if gaussian_action is None:
gaussian_action = th.from_numpy(np.arctanh(action.cpu().numpy())).to(action.device)