def __init__(self, n_classes, class_ids):
    super(CustomSoftDiceLoss, self).__init__()
    self.one_hot_encoder = One_Hot(n_classes).forward
    self.n_classes = n_classes
    self.class_ids = class_ids
def __init__(self, n_classes, class_ids):
    super(CustomSoftDiceLoss, self).__init__()
    self.one_hot_encoder = One_Hot(n_classes).forward
    self.n_classes = n_classes
    self.class_ids = class_ids