torch.nn.utils.parametrize.type_before_parametrizations

torch.nn.utils.parametrize.type_before_parametrizations(module) [source]

Return the module type before parametrizations were applied and if not, then it returns the module type.

Parameters

module (nn.Module) – module to get type of

Return type

type