def set_loss(self, loss: ClassyLoss):
    'Set loss function for task\n\n        Args:\n            loss: loss for task\n        '
    self.base_loss = loss
    return self
Set loss function for task

Args:
    loss: loss for task
def set_loss(self, loss: ClassyLoss):
    'Set loss function for task\n\n        Args:\n            loss: loss for task\n        '
    self.base_loss = loss
    return self