torch.nn.utils.parametrize.remove_parametrizations

torch.nn.utils.parametrize.remove_parametrizations(module, tensor_name, leave_parametrized=True) [исходный код]

Удалить параметризации тензора в модуле.

• Если leave_parametrized=True, module[tensor_name] будет установлено в его текущее выходное значение. В этом случае параметризация не должна изменять dtype тензора.
• Если leave_parametrized=False, module[tensor_name] будет установлено в непараметризованный тензор в module.parametrizations[tensor_name].original. Это возможно только в том случае, если параметризация зависит только от одного тензора.

Параметры:

• module (nn.Module) – модуль, из которого удаляется параметризация
• tensor_name (str) – имя удаляемой параметризации
• leave_parametrized (bool, необязательно) – оставить атрибут tensor_name параметризованным. По умолчанию: True

Возвращает:

модуль

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

Module

Вызывает исключение:

• ValueError – если module[tensor_name] не параметризован
• ValueError – если leave_parametrized=False и параметризация зависит от нескольких тензоров