D3NetBenchmark icon indicating copy to clipboard operation
D3NetBenchmark copied to clipboard

How to convert your model into ONNX file?

Open nissekl opened this issue 2 years ago • 0 comments

Hi, I used you code to train RGBNet, DepthNET and RGBDNet. And I wanna export them to ONNX file. I followed the tutoiral. I read your 3 of your network codes and find out that your input is a tuple composes of two tensors(RGB and D). So I create a Dummy tuple composed of two tensors and run the code in tutorial, but cannot export onnx file and get wierld error. image

It seems that it does scan through the network you create, but don't know why it just stop running.. Could you help me solve this problem? It really thx a lot!!!

nissekl avatar Mar 09 '23 07:03 nissekl