def __init__(self, model, reduction=torch.norm):
    super(WeightsScalarHandler, self).__init__(model, reduction)
def __init__(self, model, reduction=torch.norm):
    super(WeightsScalarHandler, self).__init__(model, reduction)