torch.nn.utils.parametrize.is_parametrized

torch.nn.utils.parametrize.is_parametrized(module, tensor_name=None)

Determine if a module has a parametrization.

Parameters

• module (nn.Module) – module to query

• tensor_name (str, optional) – name of the parameter in the module Default: None

Returns

True if module has a parametrization for the parameter named tensor_name, or if it has any parametrization when tensor_name is None; otherwise False

Return type

bool