Retrain DetectNet_v2 with QAT=True

Please provide the following information when requesting support.

• Network Type (Detectnet_v2)
• TLT Version (TAO v5.5.0)

I have trained DetectNetv2 with QAT=True initially. Then, I have pruned the model and trying to retrain the model with QAT=True again. I’m not able to retrain the model with QAT enabled.

Any specific reason why we can’t train multiple times with QAT enabled for DetectNetv2 ?

Attaching the reference here as previous discussion. Wanted to know any changes have been implemented now for supporting QAT again for DetectNetv2 while retraining

Image may be NSFW.
Clik here to view.

Please suggest. Thanks in advance

3 posts - 2 participants

Read full topic