Converting pruned model to TRT

Hey :)

I have converted plenty of models from TLT to TRT. But the issue comes when convering a pruned model. After the convertation, all the preds are simple wrong (offsets and classes are wrong). Without pruning - TRT engine works fine. With pruning, but without converting to TRT - it’s fine. So the issues comes exactly after convertation of the pruned model.

Could you give me a hand, what could this be? I use the same commands as always and it works for other models

Which network did you find the wrong convertation?

YoloV3 with ResNet18 backbone

May I ask more details about “After the convertation, all the preds are simple wrong (offsets and classes are wrong)”? Can it be reproduced with default yolo jupyter notebook?

Yeah, I’m checking the preds in the default yolo jupyter notebook myself. I can send you the models and configs, so that you can test it yourself, if you want.

Yes, please describe the details about how to reproduce the issue. If possible, you can save your jupyter notebook as an html file and attach here.