Rate this Page

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