inference Generation of tensorflow subclass model

Job ID: 33188376

Budget: ₹12,500 – ₹37,500 INR

Hi,
I need help to generate to inference to reproduce same outputs with forzen graph inference file.

my next steps would involves optimization with tensorRT. I need clear output and input files for that.
Here i have attached necessary files to generate predictions and inference generation script.

Would someone support me in generating inference(mainly involves finding subclass output and inputs)

https://github.com/deepmind/deepmind-research/tree/master/meshgraphnets

https://colab.research.google.com/drive/1lSxV93br7Ff72GYwsoCYBTG6woyR8EoL?usp=sharing

Here i added colab to link to check model train possibilities along with loading trained mdoel.
I also added script to generate inference, but output and inputs changed accordingly (encoder, processer, decoder) to produce same outputs( might need 2 inferences for encoder, processor and decoder sub class models).
Please reach me for further info.