@eleflea this seems to be a true error. We are looking into this and will post an update when it is fixed.
Do you also see this error if you load pretrained fp32 weights and then run QAT for few iterations (using the same setup)?