devit icon indicating copy to clipboard operation
devit copied to clipboard

Export to ONNX

Open crey21 opened this issue 11 months ago • 1 comments

Hi,

Is it possible to export the model used in the demo.py to ONNX ?

I've tried with the official Python methods (export and dynamo_export) with no luck, and the detectron2 methods that rely on caffe2 seem out of touch (I'd rather avoid compile my own pytorch with caffe2 enabled except if it's the only way).

Has anyone done it, or is it documented somewhere ?

(My root problem is that I cannot get it to run on linux-aarch64, even with config.MODEL.DEVICE = "cpu", whereas the same code and dependencies versions run fine on macos-arm64.)

Thanks

crey21 avatar Mar 15 '24 09:03 crey21