Open sondv2 opened 4 years ago
I think you have to modify a little bit in the torch2onnx.py to match the name in your pth file. The file is in backend/onnx_backend/torch2onnx.py. I didn't carefully handle the name in dla34, many authors will change the layer or re-define the module, it is painful to get each name of layer match.
environment: Ubuntu 18.04 GPU 1 card 1080ti When i run python network.py and got error: