remove_weight_scale

class torchelie.nn.utils.remove_weight_scale(module: torch.nn.modules.module.Module, name: str = 'weight')

Remove a weight_scale hook previously applied on getattr(module, name).