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

Fix: Set initial logging std correctly

Browse Source
This commit is contained in:
Dominik Moritz Roth 2023-08-22 01:18:44 +02:00
parent a7cbc660e0
commit 3c81a15630
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
sbBrix/common/policies.py
View File

@ -817,7 +817,7 @@ class Actor(BasePolicy):
self.mu, self.log_std = self.action_dist.proba_distribution_net( self.mu, self.log_std = self.action_dist.proba_distribution_net(
latent_dim=last_layer_dim, return_log_std=True latent_dim=last_layer_dim, return_log_std=True
) )
self._remember_log_std = log_std_init self._remember_log_std = th.Tensor([log_std_init])
# Avoid numerical issues by limiting the mean of the Gaussian # Avoid numerical issues by limiting the mean of the Gaussian
# to be in [-clip_mean, clip_mean] # to be in [-clip_mean, clip_mean]
if clip_mean > 0.0: if clip_mean > 0.0: