Finalize calibration
for name, module in model.named_modules():
     if name.endswith("_input_quantizer"):
         module.load_calib_amax()
         module.enable_quant()
If running on GPU, it needs to call .cuda() again because new tensors will be created by calibration process
model.cuda()
Keep running the quantized model