Hi Tony,
Thank you for code multitask loss for 2 tasks. If i want class work for 3 or more classes.
Does the fomula is:
1/(3sigma1^2)loss1(input1, target) +  1/(3sigma2^2) loss2(input2, target) + 1/(3sigma3^2) loss3(input3, target) + log(sigma1sigma2sigma3)
Does its right?