wang-xinyu / tensorrtx

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

Not able to deserialise the engine #1596

Open MadhavKrishna1 opened 1 month ago

MadhavKrishna1 commented 1 month ago

Env

-Xavier NX

About this repo

branch->yolov8

Your problem

You provided a code for serialsing the engine.and a codd for infer.. I want to do inference using my custom model (a layer added to yolov8).I meade .wts file from my .pt model,nad then made .engine fiel from it and it worked fine.But while doing inference it gave me erorrs as: ![Uploading Screenshot 2024-10-28 195240.png…]()