QConfig¶
- class torch.ao.quantization.qconfig.QConfig(activation, weight)[source][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))