Multiple classes for one boundingbox

[felixgodo]

Hi, Sometimes an element has two possible classifications like TITLE and DATE or DATE AND TIME.... Does anyone know how to train the model to recognise multiple classes in an element/bounding box?

[kevalmorabia97]

I think this is same as a standard multi-class multi-label classification task. You can look up online for some resources

[felixgodo]

Yes, I solved it with nn.BCEWithLogitsLoss and the performance improved also a bit