You can define multiple heads in the __init__ method of your model and use different paths in the forward method. A short example is given e.g. here, where two classifiers are used.
You can define multiple heads in the __init__ method of your model and use different paths in the forward method. A short example is given e.g. here, where two classifiers are used.