D3NetBenchmark
D3NetBenchmark copied to clipboard
How to convert your model into ONNX file?
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.

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!!!