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