wang-xinyu / tensorrtx

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

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

Open ou525 opened 1 month ago

ou525 commented 1 month 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 1 month ago

@WuxinrongY Please help here.