how to export onnx or trace pytorch model

lxtGH / OMG-Seg

OMG-LLaVA and OMG-Seg codebase

Other

905 stars 44 forks source link

how to export onnx or trace pytorch model #13

Closed adrian-wl closed 5 days ago

adrian-wl commented 6 days ago

Hello, thank you very much for the work. I want to export model to onnx format and encounter difficulties, is there a guide ? Looking forward to your reply.

adrian-wl commented 5 days ago

And can I use the demo script for testing single input? Now I need to prepare full datasets.