inference_mode

Определение

Контекстный менеджер PyTorch, отключающий autograd для ускорения инференса и снижения потребления памяти.