google-research / robotics_transformer

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

How to access the model's tf.Module object

victor-agilerobots opened this issue · comments

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