Change dropout param to use float; Fix indentation

personalrobotics · Oct 13, 2022 · d917f01 · d917f01
1 parent 2782d6c
commit d917f01
Show file tree

Hide file tree

Showing 3 changed files with 454 additions and 454 deletions.
diff --git a/d3rlpy/algos/torch/ddpg_impl.py b/d3rlpy/algos/torch/ddpg_impl.py
@@ -64,7 +64,7 @@ def __init__(
         scaler: Optional[Scaler],
         action_scaler: Optional[ActionScaler],
         reward_scaler: Optional[RewardScaler],
-        dropout : bool,
+        dropout : float,
         layernorm : bool,
     ):
         super().__init__(

diff --git a/d3rlpy/algos/torch/sac_impl.py b/d3rlpy/algos/torch/sac_impl.py
@@ -62,7 +62,7 @@ def __init__(
         scaler: Optional[Scaler],
         action_scaler: Optional[ActionScaler],
         reward_scaler: Optional[RewardScaler],
-        dropout = bool,
+        dropout = float,
         layernorm = bool,
     ):
         super().__init__(