When using the ONNX parser and plugin creator, the number of fields is zero, however when viewing the graph in NETRON, they are clearly there and populated correctly.
TensorRT Version: 126.96.36.199-1+cuda11.1
GPU Type: RTX 3090
Nvidia Driver Version: 455.23.05
CUDA Version: 11.1
CUDNN Version: 188.8.131.52-1+cuda11.1
ONNX Version: 1.6.0 (Opset 11)
Operating System + Version: ubuntu1804
Python Version (if applicable): 3.8
TensorFlow Version (if applicable): N/A
PyTorch Version (if applicable): 1.7.0
Baremetal or Container (if container which image + tag): N/A
you can find the export code in nnet_training/utilities/onnx_export.py
all the plugins are found in runtime/cerberus_net/trt_plugins, in particular I am trying to get the correlation layer working correctly, but this will apply to the others (however some have string attributes which don’t seem to export properly anyway).
I can upload an onnx graph to google drive if needed.
I feel I was reasonably close at some point to get the entire network working, just needed to implement ScatterND, but its now getting worse the more I try to fix it lol. Would be good if I did since the Correlation, Grid_sampler and ScatterND are resonably frequently used operations, once validated I can make a PR to TensorRT repo.
Steps To Reproduce
Building is the usual affair, there’s some other supplementary files such as labels and test images that you need, however you could probably just comment out those lines of code.
Reproduction-wise, I just print out the number of fields it thinks it has, which it says zero…