onnx-tensorrt
onnx-tensorrt copied to clipboard
Add Explicit Batchsize Flag
i was able to create tensorrt engine file by providing explicit batchsize flag in the trtexec tool. But when i wanted to use the onnx-tensorrt repository, i couldnt pass that flag. This is the fix that worked for me.