MiDaS
MiDaS copied to clipboard
Convert to Caffe2
Hello, has anyone successfully converted the small model to caffe2 1.6? I tried PyTorch -> ONNX -> Caffe2 but I have an error:
convert-onnx-to-caffe2 midas.onnx --output midas_predict.pb --init-net-output midas_init.pb WARNING:root:This caffe2 python run failed to load cuda module:/home/paulg/.local/lib/python3.6/site-packages/caffe2/python/../../torch/lib/libc10_cuda.so: undefined symbol: _ZN3c107Warning4warnENS_14SourceLocationERKSs,and AMD hip module:No module named 'caffe2.python.caffe2_pybind11_state_hip'.Will run in CPU only mode. Segmentation fault (core dumped)
Thank you!