NVIDIA-AI-IOT / yolo_deepstream

yolo model qat and deploy with deepstream&tensorrt
Apache License 2.0
550 stars 139 forks source link

[Yolov7_w6 - QAT] #60

Open vodanhbk95 opened 5 months ago

vodanhbk95 commented 5 months ago

According to your guide, I can do step QAT training model yolov7 on my custom dataset. But I have a question? Does your guideline work for all of Yolov7 version such as Yolov7-w6, Yolov7-l, Yolov7-x,....? Cause when i try to do QAT step by using my yolov7-w6 training model. I got this error: RuntimeError: Calibrator returned None. This usually happens when calibrator hasn't seen any tensor. Passing 'strict=False' to load_calib_amax() will ignore the error