quantize_qat
class torch.quantization.quantize_qat(model, run_fn, run_args, inplace=False)
Do quantization aware training and output a quantized model
Parameters:
      * model -- input model
  * **run_fn** -- a function for evaluating the prepared model,
    can be a function that simply runs the prepared model or a
    training loop

  * **run_args** -- positional arguments for *run_fn*

Returns:
      Quantized model.