o wZh @sdZddlmZmZddlmZddlmZmZddl m Z ddl m Z m Z gdZGdd d Zed e d Zdd edededefddZdd ededefddZdS)z;Weight Normalization from https://arxiv.org/abs/1602.07868.)AnyTypeVar) deprecated) _weight_normnorm_except_dim)Module) ParameterUninitializedParameter) WeightNorm weight_normremove_weight_normc@seZdZUeed<eed<dededdfddZdedefdd Z e e d e d dededdfd d Z deddfddZdededdfddZdS)r namedimreturnNcCs|durd}||_||_dS)N)r r)selfr rrI/var/www/auris/lib/python3.10/site-packages/torch/nn/utils/weight_norm.py__init__s zWeightNorm.__init__modulecCs.t||jd}t||jd}t|||jSN_g_v)getattrr rr)rrgvrrrcompute_weightszWeightNorm.compute_weightze`torch.nn.utils.weight_norm` is deprecated in favor of `torch.nn.utils.parametrizations.weight_norm`.)categorycCs|jD]}t|tr|j|krtd|q|durd}t||}t||}t|tr2td|j |=| |dt t |d|j | |dt |j t|||||||S)Nz>> m = weight_norm(nn.Linear(20, 40), name='weight') >>> m Linear(in_features=20, out_features=40, bias=True) >>> m.weight_g.size() torch.Size([40, 1]) >>> m.weight_v.size() torch.Size([40, 20]) )r r*)rr rrrrr Ss<r cCsV|jD]\}}t|tr |j|kr |||j|=|Sqtd|d|)aRemove the weight normalization reparameterization from a module. Args: module (Module): containing module name (str, optional): name of weight parameter Example: >>> m = weight_norm(nn.Linear(20, 40)) >>> remove_weight_norm(m) zweight_norm of 'z' not found in )ritemsr!r r r,r#)rr kr'rrrr s  r N)r)r)r))__doc__typingrrZtyping_extensionsrZtorchrrZtorch.nn.modulesrZtorch.nn.parameterrr __all__r r7r2r4r r rrrrs   B@