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, optional) – 保留属性 tensor_name 为参数化。默认值: True

返回值

模块

返回值类型

Module

抛出

• ValueError – 如果 module[tensor_name] 未参数化

• ValueError – 如果 leave_parametrized=False 且参数化依赖于多个张量