thaitc-hust/yolov9-tensorrt

torch2onnx dynamic batch

Egorundel opened this issue · 2 comments

Egorundel commented

Hello!
Can you add to the export ONNX model so that there is a dynamic batch_size?

thaitc-hust commented

@Egorundel yes. File torch2onnx.py default can export ONNX model dynamic batch_size

  • 👍1