From 98e95609130e90c05ea3da6494498c5b856fe80a Mon Sep 17 00:00:00 2001 From: Antonin RAFFIN Date: Sun, 22 Sep 2019 21:33:35 +0200 Subject: [PATCH] Remove note --- torchy_baselines/common/distributions.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/torchy_baselines/common/distributions.py b/torchy_baselines/common/distributions.py index 90258cf..e188667 100644 --- a/torchy_baselines/common/distributions.py +++ b/torchy_baselines/common/distributions.py @@ -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)