I am trying to get the RT-1 model working in PyTorch by transpiling the TensorFlow model using Ivy. I am able to load the TF model, but it's in a tensorflow.python.saved_model.load.Loader object, and I didn't find a way to extract a tf.Module object with the RT-1 model from there, which is the object I need to transpile to PyTorch using Ivy. Is there a way to get this object?
Hi, Thanks for sharing the code and dataset.
I am trying to get the RT-1 model working in PyTorch by transpiling the TensorFlow model using Ivy. I am able to load the TF model, but it's in a tensorflow.python.saved_model.load.Loader object, and I didn't find a way to extract a tf.Module object with the RT-1 model from there, which is the object I need to transpile to PyTorch using Ivy. Is there a way to get this object?
Thank you very much in advance.
Victor