torch.nn.utils.parametrize.remove_parametrizations

torch.nn.utils.parametrize.remove_parametrizations(module, tensor_name, leave_parametrized=True) [source]

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

• Если 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, optional) – оставить атрибут tensor_name параметризованным. По умолчанию: True

Возвращает

модуль

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

Модуль

Возбуждает

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