快捷方式

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

文件

查閱 PyTorch 的全面開發者文件

檢視文件

教程

獲取面向初學者和高階開發者的深入教程

檢視教程

資源

查詢開發資源並獲得問題解答

檢視資源