def METHOD_NAME(self):
    margin, nu, beta = 0.1, 0, 1
    loss_func = MarginLoss(margin=margin, nu=nu, beta=beta)
    self.assertTrue(len(loss_func.beta) == 1)
    loss_func = MarginLoss(margin=margin, nu=nu, beta=beta, learn_beta=True)
    self.assertTrue(len(loss_func.beta) == 1)
    loss_func = MarginLoss(
        margin=margin, nu=nu, beta=beta, learn_beta=True, num_classes=35
    )
    self.assertTrue(len(loss_func.beta) == 35)