Loss is stuck in Quantization Aware Training

quantization
2

Okay found the mistake, I defined the optimizer before the quantized model. Moving the optimizer line after the model declaration fixed the issue

1 Like