yes, that’s what I thought, averaging the total loss of the 5 different heads in case
we already use a similar approach of one hot encode and then split and compute the individual losses and works pretty well I would say.
I was curious to hear it there were alternatives to this or a better/optimized way to code the whole things.
thanks a lot for the prompt replies I really appreciated your help!!