instadeepai / jumanji

🕹️ A diverse suite of scalable reinforcement learning environments in JAX
https://instadeepai.github.io/jumanji
Apache License 2.0
645 stars 80 forks source link

How can i render the env when training? #242

Closed Wendyuf closed 5 months ago

Wendyuf commented 5 months ago

How can i render the env when training?

Wendyuf commented 5 months ago

How can i render the env when testing, better to save as a gift?

clement-bonnet commented 5 months ago

Hi @Wendyuf, There is a notebook in the examples folder where you can see how to animate an environment given a list of states. You can then save the animation as a GIF for instance. Please let me know if you have any questions regarding this. https://github.com/instadeepai/jumanji/blob/main/examples/load_checkpoints.ipynb

Wendyuf commented 5 months ago

Hi @Wendyuf, There is a notebook in the examples folder where you can see how to animate an environment given a list of states. You can then save the animation as a GIF for instance. Please let me know if you have any questions regarding this. https://github.com/instadeepai/jumanji/blob/main/examples/load_checkpoints.ipynb

Thank you, it helps

clement-bonnet commented 5 months ago

Great! In this case, I will close the issue. But please reopen if you have questions.