TensorRT model fails

[Jerry-Master]

After training on a custom dataset, the torch2onnx works perfectly fine but after converting to TensorRT with the exe and executing it with the exe it creates a gray screen and then halts. It does not produce any output, when using your TensorRT model everything works fine. Any clue what is happening?