I am trying to apply L1 regularization on a logistic model
class LogisticRegression(nn.Module): def __init__(self): super().__init__() s