search

google-deepmind / open_x_embodiment

Apache License 2.0
794 stars 54 forks source link

JAX/FLAX Checkpoint Release #41

Closed BakedNotFried closed 7 months ago

BakedNotFried commented 7 months ago

Are there plans to release a JAX/FLAX checkpoint that can be used by the flax checkpoint loader in the rt1_inference_example.py?

psanketi commented 7 months ago

Hello, yes, sorry we forgot to update the documentation! The checkpoint is at: gs://gdm-robotics-open-x-embodiment/open_x_embodiment_and_rt_x_oss/rt_1_x_jax/ Pl let us know if you can access it all right.

Best,

BakedNotFried commented 7 months ago

Thanks for that! I can access it and run the inference script :)