Hello,
i have added a decode_head that suppose to generate 3 outputs and i need each one of them to be trained "Joint Learning". I am actually lost between the files and I don't know what to modify in order to compute the loss and accuracy for the 3 outputs and not just single output as usual in all the models.

Can anyone please help me ?

Thanks a lot in advance
Hossam