ConvertCustomConfig¶

class torch.ao.quantization.fx.custom_config.ConvertCustomConfig[源代码][源代码]¶

用于 convert_fx() 的自定义配置。

使用示例

convert_custom_config = ConvertCustomConfig()             .set_observed_to_quantized_mapping(ObservedCustomModule, QuantizedCustomModule)             .set_preserved_attributes(["attr1", "attr2"])

classmethod from_dict(convert_custom_config_dict)[源代码][源代码]¶

从包含以下项的字典创建 ConvertCustomConfig

“observed_to_quantized_custom_module_class”: 一个嵌套字典，将量化模式映射到从观察到的模块类到量化模块类的内部映射，例如:: { “static”: {FloatCustomModule: ObservedCustomModule}, “dynamic”: {FloatCustomModule: ObservedCustomModule}, “weight_only”: {FloatCustomModule: ObservedCustomModule} } “preserved_attributes”: 即使属性在 forward 中未使用，也会保留的属性列表

此函数主要用于向后兼容，未来可能会被移除。