wang-xinyu / tensorrtx

Implementation of popular deep learning networks with TensorRT network definition API
MIT License
6.98k stars 1.77k forks source link

YOLOv9-e-convert(gelan) serialize error #1588

Open ou525 opened 3 weeks ago

ou525 commented 3 weeks ago

Env

command: ./yolov9 -s ../yolov9-e1-converted.wts yolov9-e1-converted.engine ge output: 1 ../yolov9-e1-converted.wts ge yolov9-e1-converted.engine [10/17/2024-10:44:05] [W] [TRT] The implicit batch dimension mode has been deprecated. Please create the network with NetworkDefinitionCreationFlag::kEXPLICIT_BATCH flag whenever possible. Loading weights: ../yolov9-e1-converted.wts Building engine, please wait for a while... Segmentation fault (core dumped)

But I can generate normally using the model without reparameterization. ./yolov9 -s ../yolov9-e1.wts yolov9-e1.engine e

wang-xinyu commented 3 weeks ago

@WuxinrongY Please help here.