Solved by supriyar in post #2 
              
              
                Hi @pritom-kun 
Can you clarify what you mean by performance of the model here? Are you referring to model numerics or the training time due to enabling QAT on the model? 
Regarding the numerics, the QAT step does alter the model weights assuming the model will be quantized at a later stage, so ther…