devit
devit copied to clipboard
Export to ONNX
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