sripathisony_sony:

one doubt I have trained my model with different two datasets. then I want to have two output heads.


Is the current model using a single output layer? Are these two datasets using the same number of classes? If so, what would the two heads to in your use case, i.e. would they try to predict a different number of classes?