aWTA for regression and classification?

[roydenwa]

Hi,
interesting method! Do you apply the annealed scaling only to the regression loss (positions/ mixture densities in GMMs) or to the classification loss for confidences per trajectory as well?

[yihongXU]

Hi,

only to the regression loss. But in theory, you can apply it to the classification loss.