On this page
torch.nn.utils.remove_weight_norm
torch.nn.utils.remove_weight_norm(module, name='weight')
[source]-
Removes the weight normalization reparameterization from a module.
- Parameters
- Return type
-
T_module
Example
>>> m = weight_norm(nn.Linear(20, 40)) >>> remove_weight_norm(m)
© 2024, PyTorch Contributors
PyTorch has a BSD-style license, as found in the LICENSE file.
https://pytorch.org/docs/2.1/generated/torch.nn.utils.remove_weight_norm.html