Add option for other arguments in PPO class

2024-08-08 15:07:24 +02:00 · 2024-08-08 15:07:24 +02:00 · 4287ee5a01
parent e57b2ffba4
commit 4287ee5a01
1 changed files with 1 additions and 0 deletions
--- a/rsl_rl/algorithms/ppo.py
+++ b/rsl_rl/algorithms/ppo.py
@ -30,6 +30,7 @@ class PPO:
        schedule="fixed",
        desired_kl=0.01,
        device="cpu",
+        **kwargs,
    ):
        self.device = device