search

google-deepmind / open_x_embodiment

Apache License 2.0
794 stars 54 forks source link

RT-1-X jax #55

Open uniquezhengjie opened 5 months ago

uniquezhengjie commented 5 months ago

How to convert RT-1-X jax checkpoint into tensorflow savedmodel format that can be used in Minimal_example_for_running_inference_using_RT_1_X_TF_using_tensorflow_datasets

tanakadTR commented 5 months ago

I have the same issue. I want to use a model learned by jax in Minimal_example_for_running_inference_using_RT_1_X_TF_using_tensorflow_datasets.ipynb.