convert

class torch.ao.quantization.convert(module, mapping=None, inplace=False, remove_qconfig=True, is_reference=False, convert_custom_config_dict=None, use_precomputed_fake_quant=False) [source]

Преобразует подмодули входного модуля в модули другого типа согласно mapping, вызывая метод from_float целевого класса модуля. В конце удаляет qconfig, если для remove_qconfig задано значение True.

Параметры:

• module – подготовленный и откалиброванный модуль
• mapping – словарь, сопоставляющий тип исходного модуля с типом целевого модуля; его можно переопределить, чтобы разрешить замену пользовательских модулей
• inplace – выполнять преобразования модели на месте; исходный модуль изменяется
• convert_custom_config_dict – пользовательский словарь конфигурации для функции convert
• use_precomputed_fake_quant – флаг, включающий использование предварительно вычисленной имитации квантования

# Example of convert_custom_config_dict:
convert_custom_config_dict = {
    # user will manually define the corresponding quantized
    # module class which has a from_observed class method that converts
    # observed custom module to quantized custom module
    "observed_to_quantized_custom_module_class": {
        ObservedCustomModule: QuantizedCustomModule
    }
}