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

Возвращает:

модуль

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

Модуль

Исключения:

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