Moritz Schneider

icon indicating a website link schneimo.com icon indicating an email [email protected]

icon company Robert Bosch GmbH icon location Renningen icon location Model-based Reinforcement Learning and Robot Learning @BoschResearch

Results 3 issues of Moritz Schneider

Add tf.function support

To add _tf.function_ support some parts of the main code must be heavily refactored since else the project won't work anymore or it will be slower than in eager mode.

Adding more environments

Possible environments: - MetaWorld - PyBullet

Calling JAXAgent train gets stuck if using larger image sizes (inside Ninjax)

3 comment

Hi Danijar, I am currently trying to use higher image resolutions like 256x256 for Dreamer. By simply changing the resolution e.g. for DM control suite, JAX is not able to...