torch.ao.quantization.quantize.quantize_qat

torch.ao.quantization.quantize.quantize_qat(model, run_fn, run_args, inplace=False)

Do quantization aware training and output a quantized model

参数

• 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

返回

Quantized model.

上一页

torch.ao.quantization.quantize.prepare_qat

下一页

torch.ao.quantization.quantize.quantize_dynamic