jina-ai / clip-as-service

🏄 Scalable embedding, reasoning, ranking for images and sentences with CLIP
https://clip-as-service.jina.ai
Other
12.43k stars 2.07k forks source link

how to get textual and visual onnx model #940

Open susht3 opened 11 months ago

susht3 commented 11 months ago

i train the new model, and want to get onnx models, how can i convert torch model to onnx model?

ZiniuYu commented 11 months ago

To convert a PyTorch model to ONNX, you can follow the step-by-step guide provided by PyTorch. Check out their tutorial for converting a super-resolution model as an example: PyTorch to ONNX Conversion.