tfep.nn.masked.remove_masked_weight_norm

tfep.nn.masked.remove_masked_weight_norm(module, name='weight')[source]: Remove masked weighed normalization hooks.

See also

torch.nn.utils.weight_norm.remove_weight_norm