a naive answer: If you really want to test with a single loss function for both activation functions,  what about L2 loss with one-hot vectors as target?
I’m not sure if it will give good performance though.