Modification of focal loss for it to works with mix-up augmentation?

[rtxbae]

I'm trying to train on relatively small datasets, mix-up is one way to reduce it from overfitting, but it seems like focal loss is not designed to works with label with probabilities. It seems that this line

TadTR/models/tadtr.py

Line 274 in 3af0abc

target_classes_onehot.scatter_(2, target_classes.unsqueeze(-1), 1)

specifically designed for binary classification.

Do you have any idea how to modify focal loss for label with probabilities?