Add env wrapper - Githubissues

quangr / jax-rl

jax version of ppo algorithm in mujoco enviroment, achieve SOTA(tianshou)

0 stars 0 forks source link

Open quangr opened 1 year ago

quangr commented 1 year ago

The problem for now:

Maybe should make wrappers' state frozen
envpool seem not consider handler change after reset , so now we will have two return handle by xla() and reset()