.

2023-01-28 21:47:19 +01:00
parent b86498fa9a
commit 71670782b6
2 changed files with 4 additions and 4 deletions
@@ -88,7 +88,7 @@ class ActorCriticPolicy(BasePolicy):
        activation_fn: Type[nn.Module] = nn.Tanh,
        ortho_init: bool = True,
        use_sde: bool = False,
-        log_std_init: float = 0.0,
+        std_init: float = 1.0,
        full_std: bool = True,
        sde_net_arch: Optional[List[int]] = None,
        use_expln: bool = False,
@@ -135,7 +135,7 @@ class ActorCriticPolicy(BasePolicy):
        self.features_dim = self.features_extractor.features_dim

        self.normalize_images = normalize_images
-        self.log_std_init = log_std_init
+        self.log_std_init = math.log(std_init)
        # Keyword arguments for gSDE distribution
        if dist_kwargs == None:
            dist_kwargs = {}