Why did you use sigmoid in classification head?

[samra-irshad]

I think sigmoid is not an appropriate choice to use in case of multiple classes in classification branch

[Coobiw]

In the paper, authors have said that they use sigmoid function to do the classification. They have tried softmax, but sigmoid is better probably.