.

2023-01-28 21:47:19 +01:00
parent b86498fa9a
commit 71670782b6
2 changed files with 4 additions and 4 deletions
@@ -265,7 +265,7 @@ class SACPolicy(BasePolicy):
        net_arch: Optional[Union[List[int], Dict[str, List[int]]]] = None,
        activation_fn: Type[nn.Module] = nn.ReLU,
        use_sde: bool = False,
-        log_std_init: float = -3,
+        std_init: float = 0.05,
        sde_net_arch: Optional[List[int]] = None,
        use_expln: bool = False,
        clip_mean: float = 2.0,
@@ -314,7 +314,7 @@ class SACPolicy(BasePolicy):

        sde_kwargs = {
            "use_sde": use_sde,
-            "log_std_init": log_std_init,
+            "log_std_init": math.log(std_init),
            "use_expln": use_expln,
            "clip_mean": clip_mean,
        }