about the variance of classification task
ShellingFord221 opened this issue · 0 comments
ShellingFord221 commented
Hi, when modeling aleatoric uncertainty, why is pred_var, predicted logit values and variance's shape: (N, C + 1)? Shouldn't it be (N, C * 2)?