From pixels?

[slerman12]

Is there a way to get the states as image frames?

[erikfrey]

Hi Sam,

We're pushing an update with support for rendering states on Tuesday or Wednesday. Stay tuned!

[erikfrey]

OK, rendering to pixels is now available, but not yet documented. If you'd like to try it, please check out:

https://github.com/google/brax/blob/main/brax/io/image.py

Please keep in mind this is CPU rendering, so better for eval rendering and other programmatic use cases, rather than training. We will move to GPU/TPU rendering in the future, which should be suitable for training.

In the coming days we'll update our colabs with an example of how to use it.

[slerman12]

Ah, GPU training is what I need. But thank you!

[erikfrey]

Closing this as I think it's addressed, but please do re-open if you have any other questions. We are tracking the request to render using accelerators at #108