Hi,
You mentioned the usage as:
usage
is_regression = torch.Tensor([True, True, False]) # True: Regression/MeanSquaredErrorLoss, False: Classification/CrossEntropyLoss
multitaskloss_instance = MultiTaskLoss(is_regression)
So in case of classification problem I should put
is_regression = False
can clarify it a bit ?