the mapping argument to prepare_qat (https://github.com/pytorch/pytorch/blob/733b8c23c436d906125c20f0a64692bf57bce040/torch/quantization/quantize.py#L289 10) can be used to customize which layer’s you’d like to quantize