chengming
/
rsl_rl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add option for other arguments in PPO class

This commit is contained in:
kaixi287 2024-08-08 15:07:24 +02:00
parent e57b2ffba4
commit 4287ee5a01
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
rsl_rl/algorithms/ppo.py
View File

@ -30,6 +30,7 @@ class PPO:
schedule="fixed", schedule="fixed",
desired_kl=0.01, desired_kl=0.01,
device="cpu", device="cpu",
**kwargs,
): ):
self.device = device self.device = device