how can i export pretraind model to onnx model?

eric-yyjau / pytorch-superpoint

Superpoint Implemented in PyTorch: https://arxiv.org/abs/1712.07629

MIT License

784 stars 164 forks source link

Open jcyhcs opened 2 years ago

jcyhcs commented 2 years ago

hi,professor: how can export pretrained model to onnx format? please help!

eric-yyjau commented 2 years ago

Hi @jcyhcs,

(Thanks but I'm not a professor though). This tutorial may help (https://pytorch.org/tutorials/advanced/super_resolution_with_onnxruntime.html). This repo does not cover exporting to onnx but I imagine the general pytorch method works.