quantize_qat

class torch.ao.quantization.quantize_qat(model, run_fn, run_args, inplace=False) [исходный код]

Выполняет обучение с учетом квантования и возвращает квантованную модель

Параметры:

model – входная модель
run_fn – функция для оценки подготовленной модели; это может быть функция, которая просто запускает подготовленную модель, или цикл обучения
run_args – позиционные аргументы для run_fn

Возвращает:

Квантованная модель.