enable_observer

class torch.ao.quantization.fake_quantize.enable_observer(mod) [исходный код]

Включает наблюдение для этого модуля.

Включает наблюдение для этого модуля, если это применимо. Пример использования:

# model is any PyTorch model
model.apply(torch.ao.quantization.enable_observer)