I didn’t quite get your reply sorry but what I’m trying to convey is that we can directly use torch2trt and I’m using TensorRT’s latest version plus torch’s latest version and the model’s using all the supported operations of tensorrt so I would assume we don’t need any custom plugins for this. I just need to understand what is the exact meaning of the above error which is clearly related to engine of tensorrt and I provided on which line the error raised. I tried all my ways and couldn’t get any help, please let me know if you have any solution for this?
The error message indicates that the scales of the normalization layer in the model are not compatible with the dimensions of the input tensor. This may happen if the model was trained on a different image size than the input tensor.
Please check the image size that the model was trained on and resize the input tensor to the same size.
We obtained the following output after adding the following: