torch.nn.utils.parametrize.is_parametrized

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

Returns True if module has an active parametrization.

If the argument tensor_name is specified, returns True if module[tensor_name] is parametrized.

Parameters

• module (nn.Module) – module to query

• tensor_name (str, optional) – attribute in the module to query Default: None

Return type

bool