search

Deci-AI / super-gradients

Easily train or fine-tune SOTA computer vision models with one open source training library. The home of Yolo-NAS.
https://www.supergradients.com
Apache License 2.0
4.59k stars 510 forks source link

Issue with inferencing on a tensorrt model (converted from yolo_nas_s) #1200

Closed Atm-ninetyseven closed 1 year ago

Atm-ninetyseven commented 1 year ago

💡 Your Question

I am experiencing difficulties when attempting to perform inference on a TensorRT model. I have come to the conclusion that the fine-tuning hyper-parameters listed under ''Quantization Aware Training Yolo NAS on Custom Dataset Notebook' are not enough to retrieve a well quantized tensorRT model for the given dataset. Could you provide more documentation on hyper-parameter tuning relating to quantization aware training?

Thank you!

ChuRuaNh0 commented 1 year ago

https://github.com/Deci-AI/super-gradients/issues/1288

harpreetsahota204 commented 1 year ago

Hey @Atm-ninetyseven @ChuRuaNh0

Thanks for opening an issue for SG. I'm gathering some feedback on SuperGradients and YOLO-NAS.

Would you be down for a quick call to chat about your experience?

If a call doesn't work for you, no worries. I've got a short survey you could fill out: https://bit.ly/sgyn-feedback.

I know you’re super busy, but your input will help us shape the direction of SuperGradients and make it as useful as possible for you.

I appreciate your time and feedback. Let me know what works for you.

Cheers,

Harpreet