I am creating a custom loss function that uses two metric
def loss_fn(preds,target,a,b) :
          loss1 = loss_func1(preds,target)
          loss2 = loss_func2(preds,target)
          loss = a*loss1 + b*loss2
          return loss

How do i make variable a and b, in the above function, learnable in the training loop