Export models to Onnx

Question

Export models to Onnx

Closed this issue 13 days ago · 3 comments

Been trying to export the ppo models to an onnx model but keep running into errors with how it might be implemented. Have any of you exported the trained models to onnx format?

Answer 1 · 2024-09-18T15:17:53.000Z

Hi @zcase, no I am afraid I haven't, and to my knowledge no one else has. I am not familiar with onnx at all unfortunately. What is the benefit of onnx, and why are you using it?

Answer 2 · 2024-09-18T15:37:59.000Z

@Federico-PizarroBejarano Onnx is a model format that allows you to transfer in between different frameworks like TensorFlow and Torch. I have found it really helpful when transitioning things to c++. Torch has a way to export to onnx models really easily, however when I tried it with the PPO controller models. It looked like there were some errors. I did modify somethings and was able to get an export but it may have broken the rest of the algorithm when doing training.

Answer 3 · 2024-09-18T15:45:52.000Z

Rip I am sorry to hear that, but I hope you get it to work! It seems very cool. If more people are interested, we may try to add support for onnx in the gym at some point then. Good luck!