評價此頁

QConfig#

class torch.ao.quantization.qconfig.QConfig(activation, weight)[source]#

透過為啟用和權重分別提供設定(觀察器類)來描述如何量化一個層或網路的一部分。

請注意,QConfig 需要包含觀察器 **類**(如 MinMaxObserver)或一個在呼叫時返回例項的可呼叫物件,而不是具體的觀察器例項本身。量化準備函式將為每個層多次例項化觀察器。

觀察器類通常具有合理的預設引數,但可以使用 with_args 方法(其行為類似於 functools.partial)進行覆蓋。

my_qconfig = QConfig(
    activation=MinMaxObserver.with_args(dtype=torch.qint8),
    weight=default_observer.with_args(dtype=torch.qint8),
)