From 4287ee5a01417b129879c8ea6951c7a98e3da95e Mon Sep 17 00:00:00 2001 From: kaixi287 Date: Thu, 8 Aug 2024 15:07:24 +0200 Subject: [PATCH] Add option for other arguments in PPO class --- rsl_rl/algorithms/ppo.py | 1 + 1 file changed, 1 insertion(+) diff --git a/rsl_rl/algorithms/ppo.py b/rsl_rl/algorithms/ppo.py index ffe5814..a589b76 100644 --- 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