Revert "Deprecate reparameterization module (#1316)" (#1319)

This reverts commit adbe075a.

apex/reparameterization/__init__.py

-import warnings
-
 from .weight_norm import WeightNorm
 from .reparameterization import Reparameterization
 
-
 def apply_weight_norm(module, name='', dim=0, hook_child=True):
     r"""
     Applies weight normalization to a parameter in the given module.
@@ -31,7 +28,7 @@ def apply_weight_norm(module, name='', dim=0, hook_child=True):
         module (nn.Module): containing module
         name (str, optional): name of weight parameter
         dim (int, optional): dimension over which to compute the norm
-        hook_child (boolean, optional): adds reparameterization hook to direct parent of the
+        hook_child (boolean, optional): adds reparameterization hook to direct parent of the 
             parameters. If False, it's added to `module` instead. Default: True
 
     Returns:
@@ -47,7 +44,6 @@ def apply_weight_norm(module, name='', dim=0, hook_child=True):
         torch.Size([40, 20])
 
     """
-    warnings.warn("`apply_weight_norm` will be removed by the end of June, 2022.", FutureWarning)
     return apply_reparameterization(module, reparameterization=WeightNorm, hook_child=hook_child,
                                     name=name, dim=dim)
 
@@ -62,7 +58,6 @@ def remove_weight_norm(module, name='', remove_all=False):
         >>> m = apply_weight_norm(nn.Linear(20, 40))
         >>> remove_weight_norm(m)
     """
-    warnings.warn("`remove_weight_norm` will be removed by the end of June, 2022.", FutureWarning)
     return remove_reparameterization(module, reparameterization=WeightNorm,
                                     name=name, remove_all=remove_all)
 
@@ -77,7 +72,7 @@ def apply_reparameterization(module, reparameterization=None, name='', dim=0, ho
         reparameterization (Reparameterization): reparamaterization class to apply
         name (str, optional): name of weight parameter
         dim (int, optional): dimension over which to perform reparameterization op
-        hook_child (boolean, optional): adds reparameterization hook to direct parent of the
+        hook_child (boolean, optional): adds reparameterization hook to direct parent of the 
             parameters. If False, it's added to `module` instead. Default: True
 
     Returns:
@@ -89,10 +84,6 @@ def apply_reparameterization(module, reparameterization=None, name='', dim=0, ho
         Linear (20 -> 40)
 
     """
-    warnings.warn(
-        "`apply_reparameterization` will be removed by the end of June, 2022.",
-        FutureWarning,
-    )
     assert reparameterization is not None
     if name != '':
         Reparameterization.apply(module, name, dim, reparameterization, hook_child)
@@ -116,10 +107,6 @@ def remove_reparameterization(module, reparameterization=Reparameterization,
         >>> m = apply_reparameterization(nn.Linear(20, 40),WeightNorm)
         >>> remove_reparameterization(m)
     """
-    warnings.warn(
-        "`remove_reparameterization` will be removed by the end of June, 2022.",
-        FutureWarning,
-    )
     if name != '' or remove_all:
         to_remove = []
         for k, hook in module._forward_pre_hooks.items():

apex/reparameterization/weight_norm.py

-import warnings
-
 import torch
 from torch.nn.parameter import Parameter
 from ..fp16_utils import Fused_Weight_Norm