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), )