def METHOD_NAME(self):
    """Remove weight normalization module from all of the layers."""
    def _remove_weight_norm(m: torch.nn.Module):
        try:
            logging.debug(f"Weight norm is removed from {m}.")
            torch.nn.utils.METHOD_NAME(m)
        except ValueError:  # this module didn't have weight norm
            return
    self.apply(_remove_weight_norm)