prepare

class torch.ao.quantization.prepare(model, inplace=False, allow_list=None, observer_non_leaf_module_list=None, prepare_custom_config_dict=None) [исходный код]

Подготавливает копию модели для калибровки квантования или обучения с учетом квантования.

Конфигурацию квантования необходимо заранее назначить отдельным подмодулям в атрибуте .qconfig.

К модели будут добавлены модули наблюдателей или фиктивного квантования, а qconfig будет распространен.

Параметры:

• model – входная модель, которая будет изменена на месте
• inplace – выполнять преобразования модели на месте; исходный модуль будет изменен
• allow_list – список модулей, поддерживающих квантование
• observer_non_leaf_module_list – список нелистовых модулей, к которым нужно добавить наблюдатель
• prepare_custom_config_dict – словарь конфигурации пользовательских настроек для функции prepare

# Example of prepare_custom_config_dict:
prepare_custom_config_dict = {
    # user will manually define the corresponding observed
    # module class which has a from_float class method that converts
    # float custom module to observed custom module
    "float_to_observed_custom_module_class": {CustomModule: ObservedCustomModule}
}