Puh, CrossEentropyLoss should be fine, it the target classes are equiprobable.
In other words, if you have the same number of samples for each class the criterion should be good enough?
I would try it and see what happens.
Let me know, if you could generate some nice images!