FuseCustomConfig

class torch.ao.quantization.fx.custom_config.FuseCustomConfig [исходный код]

Пользовательская конфигурация для fuse_fx().

Пример использования:

fuse_custom_config = FuseCustomConfig().set_preserved_attributes(
    ["attr1", "attr2"]
)

classmethod from_dict(fuse_custom_config_dict) [исходный код]

Создаёт ConvertCustomConfig из словаря со следующими элементами:

«preserved_attributes»: список атрибутов, которые сохраняются, даже если они не используются в forward

Эта функция предназначена главным образом для обратной совместимости и может быть удалена в будущем.

Тип возвращаемого значения:

FuseCustomConfig

set_preserved_attributes(attributes) [исходный код]

Задаёт имена атрибутов, которые будут сохраняться в модуле графа, даже если они не используются в методе forward модели.

Тип возвращаемого значения:

FuseCustomConfig

to_dict() [исходный код]

Преобразует этот FuseCustomConfig в словарь с элементами, описанными в from_dict().

Тип возвращаемого значения:

dict[str, Any]